Skip to content

Commit b96ac05

Browse files
committed
apply styler
1 parent 46b5929 commit b96ac05

File tree

1 file changed

+26
-26
lines changed

1 file changed

+26
-26
lines changed

tests/testthat/test-double_ml_pliv_partial_functional_initializer_IVtype.R

Lines changed: 26 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)