tests/testthat/test-select.R

In krlmlr/fstplyr: A 'dplyr' Interface to 'fst'

context("select")

test_that("can select", {
  src <- src_fst("src")
  tbl <- tbl(src, "iris")

  expect_identical(tbl %>% select(1:5) %>% collect(), tbl %>% collect())
  expect_identical(tbl %>% select(-Species) %>% collect(), tbl %>% collect() %>% select(-Species))
  expect_identical(tbl %>% select(Species) %>% head() %>% collect(), tbl %>% collect() %>% select(Species) %>% head())
  expect_identical(tbl %>% select() %>% collect(), tbl %>% collect() %>% select())
  expect_identical(tbl %>% select(a = Species) %>% collect(), tbl %>% collect() %>% select(a = Species))
  expect_identical(
    tbl %>%
      select(-starts_with("Petal")) %>%
      select(Species, everything()) %>%
      collect(),
    tbl %>%
      collect() %>%
      select(Species, starts_with("Sepal"))
  )
})

test_that("can rename", {
  src <- src_fst("src")
  tbl <- tbl(src, "iris")

  expect_identical(tbl %>% rename(a = Species) %>% collect(), tbl %>% collect() %>% select(everything(), a = Species))
  expect_identical(
    tbl %>%
      rename(a = Species, b = Petal.Width) %>%
      rename(c = a, d = Petal.Length) %>%
      collect(),
    tbl %>%
      collect() %>%
      select(Sepal.Length, Sepal.Width, d = Petal.Length, b = Petal.Width, c = Species)
  )
})

test_that("colnames work after select", {
  src <- src_fst("src")
  tbl <- tbl(src, "iris")

  expect_identical(tbl %>% select(a = Species) %>% colnames(), "a")
})