Update test_abstract_test_case.py

Author: christopher-wild

tests/generation_tests/test_abstract_test_case.py

@@ -33,7 +33,7 @@ def test_generate_concrete_test_cases(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={scenario.treatment_variables[self.X1.name].z3 > self.X1.z3},
-            treatment_variables={self.X1},
+            treatment_variables=self.X1,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )
@@ -47,7 +47,7 @@ def test_str(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={scenario.treatment_variables[self.X1.name].z3 > self.X1.z3},
-            treatment_variables={self.X1},
+            treatment_variables=self.X1,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )
@@ -61,7 +61,7 @@ def test_datapath(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={scenario.treatment_variables[self.X1.name].z3 > self.X1.z3},
-            treatment_variables={self.X1},
+            treatment_variables=self.X1,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )
@@ -73,7 +73,7 @@ def test_generate_concrete_test_cases_with_constraints(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={scenario.treatment_variables[self.X1.name].z3 > self.X1.z3},
-            treatment_variables={self.X1},
+            treatment_variables=self.X1,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )
@@ -87,7 +87,7 @@ def test_generate_concrete_test_cases_with_effect_modifiers(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={scenario.treatment_variables[self.X1.name].z3 > self.X1.z3},
-            treatment_variables={self.X1},
+            treatment_variables=self.X1,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers={self.X2},
         )
@@ -101,7 +101,7 @@ def test_generate_concrete_test_cases_rct(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={scenario.treatment_variables[self.X1.name].z3 > self.X1.z3},
-            treatment_variables={self.X1},
+            treatment_variables=self.X1,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )
@@ -116,7 +116,7 @@ def test_infeasible_constraints(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={scenario.treatment_variables[self.X1.name].z3 > self.X1.z3},
-            treatment_variables={self.X1},
+            treatment_variables=self.X1,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )
@@ -135,7 +135,7 @@ def test_feasible_constraints(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={scenario.treatment_variables[self.X1.name].z3 > self.X1.z3},
-            treatment_variables={self.X1},
+            treatment_variables=self.X1,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )