Add estimator_params to test

christopher-wild · christopher-wild · commit b372d2260d0d · 2023-08-08T09:06:14.000+01:00
diff --git a/tests/testing_tests/test_estimators.py b/tests/testing_tests/test_estimators.py
@@ -394,7 +394,7 @@ def test_program_15_no_interaction_ate_calculated(self):
         # for term_to_square in terms_to_square:
 
         ate, [ci_low, ci_high] = linear_regression_estimator.estimate_ate_calculated(
-            adjustment_config={k: self.nhefs_df.mean()[k] for k in covariates}
+            estimator_params={"adjustment_config": {k: self.nhefs_df.mean()[k] for k in covariates}}
         )
         self.assertEqual(round(ate, 1), 3.5)
         self.assertEqual([round(ci_low, 1), round(ci_high, 1)], [1.9, 5])

Original file line number	Diff line number	Diff line change
`@@ -394,7 +394,7 @@ def test_program_15_no_interaction_ate_calculated(self):`
`394`	`394`	`# for term_to_square in terms_to_square:`
`395`	`395`
`396`	`396`	`ate, [ci_low, ci_high] = linear_regression_estimator.estimate_ate_calculated(`
`397`		`- adjustment_config={k: self.nhefs_df.mean()[k] for k in covariates}`
	`397`	`+ estimator_params={"adjustment_config": {k: self.nhefs_df.mean()[k] for k in covariates}}`
`398`	`398`	`)`
`399`	`399`	`self.assertEqual(round(ate, 1), 3.5)`
`400`	`400`	`self.assertEqual([round(ci_low, 1), round(ci_high, 1)], [1.9, 5])`