tests/testthat/test-srckeep.r

In xiaodaigh/disk.frame: Larger-than-RAM Disk-Based Data Manipulation Framework

context("test-srckeep")

setup({
  b = data.frame(a = 51:150, b = 1:100)
  as.disk.frame(b, file.path(tempdir(), "tmp_srckeep.df"), nchunks = 5, overwrite = TRUE)
})

test_that("testing srckeep", {
  b = disk.frame(file.path(tempdir(), "tmp_srckeep.df"))
  b1 = b %>% srckeep("a")
  expect_equal(ncol(b1 %>% collect), 1)
  expect_equal(colnames(b1 %>% collect), "a")
})

teardown({
  fs::dir_delete(file.path(tempdir(), "tmp_srckeep.df"))
})