adjust tests

bips-hb · Jun 3, 2024 · ea9526c · ea9526c
1 parent 949b026
commit ea9526c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/testthat/test_expct.R b/tests/testthat/test_expct.R
@@ -40,14 +40,14 @@ test_that("expct works with partial sample", {
   res <- expct(psi, evidence = evi, parallel = FALSE)
   expect_s3_class(res, "data.frame")
   expect_equal(nrow(res), nrow(evi))
-  expect_equal(ncol(res), ncol(iris))
-  expect_equal(colnames(res), colnames(iris))
+  expect_equal(ncol(res), ncol(iris) - ncol(evi))
+  expect_equal(colnames(res), colnames(iris)[3:4])
 
   # Petal.* and Species missing
   evi <- iris[1:3, 1:2]
   res <- expct(psi, evidence = evi, parallel = FALSE)
   expect_s3_class(res, "data.frame")
   expect_equal(nrow(res), nrow(evi))
-  expect_equal(ncol(res), ncol(iris))
-  expect_equal(colnames(res), colnames(iris))
+  expect_equal(ncol(res), ncol(iris) - ncol(evi))
+  expect_equal(colnames(res), colnames(iris)[3:5])
 })