tests/testthat/test-full_join.R

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

context("test-full_join")

setup({
  a = data.frame(a = 1:100, b = 1:100)
  b = data.frame(a = 51:150, b = 1:100)
  d = data.frame(a = 1:50, b = 1:50)
  
  as.disk.frame(a, file.path(tempdir(), "tmp_a_fj.df"), nchunks = 4, overwrite = T)
  as.disk.frame(b, file.path(tempdir(), "tmp_b_fj.df"), nchunks = 5, overwrite = T)
  as.disk.frame(d, file.path(tempdir(), "tmp_d_fj.df"), overwrite = T)
})

test_that("testing full_join where right is data.frame", {
  a = disk.frame(file.path(tempdir(), "tmp_a_fj.df"))
  b = disk.frame(file.path(tempdir(), "tmp_b_fj.df"))
  d = disk.frame(file.path(tempdir(), "tmp_d_fj.df"))
  bc = collect(b)
  dc = collect(d)
  
  abc <- full_join(a, bc, by = "a") %>% collect
  expect_equal(nrow(abc), 150)
  
  abc0 <- full_join(a, bc, by = c("a","b")) %>% collect
  expect_equal(nrow(abc0), 200)
  
  abc100 <- full_join(a, bc, by = "b") %>% collect
  expect_equal(nrow(abc100), 100)
  
  abd50 <- full_join(a, dc, by = "b") %>% collect
  expect_equal(nrow(abd50), 100)
})

test_that("testing full_join where right is disk.frame", {
  a = disk.frame(file.path(tempdir(), "tmp_a_fj.df"))
  b = disk.frame(file.path(tempdir(), "tmp_b_fj.df"))
  d = disk.frame(file.path(tempdir(), "tmp_d_fj.df"))
  
  expect_warning({
    ab <- full_join(a, b, by = "a", merge_by_chunk_id = F) %>% collect
    })
  expect_equal(nrow(ab), 150)
  
  expect_warning({ab0 = full_join(a, b, by = c("a","b"), merge_by_chunk_id = F) %>% collect})
  expect_equal(nrow(ab0), 200)
  
  expect_warning({ab100 = full_join(a, b, by = "b", merge_by_chunk_id = F) %>% collect})
  expect_equal(nrow(ab100), 100)
  
  expect_warning({ad50 = full_join(a, d, by = "b", merge_by_chunk_id = F) %>% collect})
  expect_equal(nrow(ad50), 100)
})


teardown({
  fs::dir_delete(file.path(tempdir(), "tmp_a_fj.df"))
  fs::dir_delete(file.path(tempdir(), "tmp_b_fj.df"))
  fs::dir_delete(file.path(tempdir(), "tmp_d_fj.df"))
})