fix test

christopher-wild · christopher-wild · commit 8cbf150fa59a · 2023-08-07T16:31:40.000+01:00
diff --git a/tests/testing_tests/test_estimators.py b/tests/testing_tests/test_estimators.py
@@ -392,8 +392,9 @@ def test_program_15_no_interaction_ate_calculated(self):
         )
         # terms_to_square = ["age", "wt71", "smokeintensity", "smokeyrs"]
         # for term_to_square in terms_to_square:
+
         ate, [ci_low, ci_high] = linear_regression_estimator.estimate_ate_calculated(
-            {k: self.nhefs_df.mean()[k] for k in covariates}
+            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
`@@ -392,8 +392,9 @@ def test_program_15_no_interaction_ate_calculated(self):`
`392`	`392`	`)`
`393`	`393`	`# terms_to_square = ["age", "wt71", "smokeintensity", "smokeyrs"]`
`394`	`394`	`# for term_to_square in terms_to_square:`
	`395`	`+`
`395`	`396`	`ate, [ci_low, ci_high] = linear_regression_estimator.estimate_ate_calculated(`
`396`		`- {k: self.nhefs_df.mean()[k] for k in covariates}`
	`397`	`+ adjustment_config={k: self.nhefs_df.mean()[k] for k in covariates}`
`397`	`398`	`)`
`398`	`399`	`self.assertEqual(round(ate, 1), 3.5)`
`399`	`400`	`self.assertEqual([round(ci_low, 1), round(ci_high, 1)], [1.9, 5])`