@@ -24,40 +24,40 @@ patrick::with_parameters_test_that("Unit tests for PLIV (partialX functional ini
2424 df = data_pliv $ df
2525 Xnames = names(df )[names(df ) %in% c(" y" , " d" , " z" , " z2" ) == FALSE ]
2626 data_ml = double_ml_data_from_data_frame(df ,
27- y_col = " y" ,
28- d_cols = " d" , x_cols = Xnames , z_cols = " z" )
29-
27+ y_col = " y" ,
28+ d_cols = " d" , x_cols = Xnames , z_cols = " z" )
29+
3030 # Partial out X (default PLIV)
3131 set.seed(3141 )
3232 double_mlpliv_obj = DoubleMLPLIV $ new(data_ml ,
33- n_folds = 5 ,
34- ml_l = learner_pars $ ml_l $ clone(),
35- ml_m = learner_pars $ ml_m $ clone(),
36- ml_r = learner_pars $ ml_r $ clone(),
37- ml_g = learner_pars $ ml_g $ clone(),
38- dml_procedure = dml_procedure ,
39- score = score )
40-
33+ n_folds = 5 ,
34+ ml_l = learner_pars $ ml_l $ clone(),
35+ ml_m = learner_pars $ ml_m $ clone(),
36+ ml_r = learner_pars $ ml_r $ clone(),
37+ ml_g = learner_pars $ ml_g $ clone(),
38+ dml_procedure = dml_procedure ,
39+ score = score )
40+
4141 double_mlpliv_obj $ fit()
4242 theta_obj = double_mlpliv_obj $ coef
4343 se_obj = double_mlpliv_obj $ se
44-
44+
4545 # Partial out X
4646 set.seed(3141 )
4747 double_mlpliv_partX = DoubleMLPLIV.partialX(data_ml ,
48- n_folds = 5 ,
49- ml_l = learner_pars $ ml_l $ clone(),
50- ml_m = learner_pars $ ml_m $ clone(),
51- ml_r = learner_pars $ ml_r $ clone(),
52- ml_g = learner_pars $ ml_g $ clone(),
53- dml_procedure = dml_procedure ,
54- score = score )
55-
56- double_mlpliv_partX $ fit()
57- theta_partX = double_mlpliv_partX $ coef
58- se_partX = double_mlpliv_partX $ se
59-
60- expect_equal(theta_partX , theta_obj , tolerance = 1e-8 )
61- expect_equal(se_partX , se_obj , tolerance = 1e-8 )
48+ n_folds = 5 ,
49+ ml_l = learner_pars $ ml_l $ clone(),
50+ ml_m = learner_pars $ ml_m $ clone(),
51+ ml_r = learner_pars $ ml_r $ clone(),
52+ ml_g = learner_pars $ ml_g $ clone(),
53+ dml_procedure = dml_procedure ,
54+ score = score )
55+
56+ double_mlpliv_partX $ fit()
57+ theta_partX = double_mlpliv_partX $ coef
58+ se_partX = double_mlpliv_partX $ se
59+
60+ expect_equal(theta_partX , theta_obj , tolerance = 1e-8 )
61+ expect_equal(se_partX , se_obj , tolerance = 1e-8 )
6262 }
6363)
0 commit comments