add some basic unit tests for the export of estimated models

Author: MalteKurz

tests/testthat/test-double_ml_active_bindings.R

@@ -40,6 +40,9 @@ test_that("Not setable fields", {
   msg = "can't set field learner"
   expect_error(dml_plr$learner <- "abc",
     regexp = msg)
+  msg = "can't set field models"
+  expect_error(dml_plr$models <- 5,
+    regexp = msg)
   msg = "can't set field n_folds"
   expect_error(dml_plr$n_folds <- 5,
     regexp = msg)

tests/testthat/test-double_ml_plr_export_preds.R

@@ -46,8 +46,9 @@ patrick::with_parameters_test_that("Unit tests for for the export of predictions
       n_folds = n_folds,
       score = score)
     set.seed(3141)
-    double_mlplr_obj$fit(store_predictions = TRUE)
+    double_mlplr_obj$fit(store_predictions = TRUE, store_models = TRUE)
 
+    browser()
     set.seed(3141)
     Xnames = names(df)[names(df) %in% c("y", "d", "z") == FALSE]
     indx = (names(df) %in% c(Xnames, "y"))
@@ -99,14 +100,20 @@ patrick::with_parameters_test_that("Unit tests for for the export of predictions
       expect_equal(as.vector(double_mlplr_obj$predictions$ml_g),
         as.vector(preds_g$response),
         tolerance = 1e-8)
+      expect_class(double_mlplr_obj$models$ml_g$d[[1]][[1]],
+                   "LearnerRegr")
     }
 
     expect_equal(as.vector(double_mlplr_obj$predictions$ml_l),
       as.vector(preds_l$response),
       tolerance = 1e-8)
+    expect_class(double_mlplr_obj$models$ml_l$d[[1]][[1]],
+                 "LearnerRegr")
 
     expect_equal(as.vector(double_mlplr_obj$predictions$ml_m),
       as.vector(preds_m$response),
       tolerance = 1e-8)
+    expect_class(double_mlplr_obj$models$ml_m$d[[1]][[1]],
+                 "LearnerRegr")
   }
 )