Skip to content

Commit 2131be7

Browse files
author
AndrewC19
committed
Fixed incorrect test assertion
1 parent 9c67392 commit 2131be7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/testing_tests/test_causal_test_engine.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,7 @@ def test_execute_test_observational_linear_regression_estimator_direct_effect(se
179179
)
180180
causal_test_engine.identification(causal_test_case)
181181
self.minimal_adjustment_set = causal_test_engine.minimal_adjustment_set
182+
print(self.minimal_adjustment_set)
182183
# 6. Easier to access treatment and outcome values
183184
self.treatment_value = 1
184185
self.control_value = 0
@@ -189,7 +190,7 @@ def test_execute_test_observational_linear_regression_estimator_direct_effect(se
189190
('C',),
190191
causal_test_engine.scenario_execution_data_df)
191192
causal_test_result = causal_test_engine.execute_test(estimation_model, causal_test_case)
192-
self.assertAlmostEqual(causal_test_result.ate, 0, delta=1e-10)
193+
self.assertAlmostEqual(causal_test_result.ate, 4, delta=1e-10)
193194

194195
def test_execute_test_observational_linear_regression_estimator_risk_ratio(self):
195196
""" Check that executing the causal test case returns the correct results for dummy data using a linear

0 commit comments

Comments
 (0)