Update causal_test_case.py and related tests

Author: christopher-wild

causal_testing/testing/causal_test_case.py

@@ -57,7 +57,7 @@ def get_treatment_variables(self):
 
     def get_outcome_variables(self):
         """Return a list of the outcome variables (as strings) for this causal test case."""
-        return [v.name for v in self.outcome_variables]
+        return [self.outcome_variable.name]
 
     def get_control_values(self):
         """Return a list of the control values for each treatment variable in this causal test case."""
@@ -70,7 +70,8 @@ def get_treatment_values(self):
     def __str__(self):
         treatment_config = {k.name: v for k, v in self.treatment_input_configuration.items()}
         control_config = {k.name: v for k, v in self.control_input_configuration.items()}
+        outcome_variable = {self.outcome_variable}
         return (
             f"Running {treatment_config} instead of {control_config} should cause the following "
-            f"changes to {self.outcome_variable}: {self.expected_causal_effect}."
+            f"changes to {outcome_variable}: {self.expected_causal_effect}."
         )

tests/testing_tests/test_causal_test_case.py

@@ -6,6 +6,7 @@
 from causal_testing.specification.causal_dag import CausalDAG
 from causal_testing.testing.causal_test_case import CausalTestCase
 from causal_testing.testing.causal_test_outcome import ExactValue
+from causal_testing.testing.base_causal_test import BaseCausalTest
 
 class TestCausalTestEngineObservational(unittest.TestCase):
     """ Test the CausalTestEngine workflow using observational data.
@@ -33,11 +34,13 @@ def setUp(self) -> None:
 
         # 3. Create an intervention and causal test case
         self.expected_causal_effect = ExactValue(4)
+        self.base_causal_test = BaseCausalTest(A, C)
         self.causal_test_case = CausalTestCase(
-            control_input_configuration={A: 0},
+            base_causal_test=self.base_causal_test,
             expected_causal_effect=self.expected_causal_effect,
-            treatment_input_configuration={A: 1},
-            outcome_variables={C})
+            control_value=0,
+            treatment_value=1
+        )
 
     def test_get_treatment_variables(self):
         self.assertEqual(self.causal_test_case.get_treatment_variables(), ["A"])