Tests now use new test result layout

Author: christopher-wild

examples/poisson/run_causal_tests.py

@@ -160,8 +160,8 @@ def add_modelling_assumptions(self, estimation_model: Estimator):
 if __name__ == "__main__":
     args = MyJsonUtility.get_args()
     json_utility = MyJsonUtility(args.log_path)  # Create an instance of the extended JsonUtility class
-    json_utility.set_path(args.json_path, args.dag_path,
-                          args.data_path)  # Set the path to the data.csv, dag.dot and causal_tests.json file
+    json_utility.set_paths(args.json_path, args.dag_path,
+                           args.data_path)  # Set the path to the data.csv, dag.dot and causal_tests.json file
 
     # Load the Causal Variables into the JsonUtility class ready to be used in the tests
     json_utility.set_variables(inputs, outputs, metas)

tests/testing_tests/test_causal_test_outcome.py

@@ -1,18 +1,21 @@
 import unittest
 from causal_testing.testing.causal_test_outcome import ExactValue, SomeEffect
 from causal_testing.testing.causal_test_result import CausalTestResult, TestValue
-
+from causal_testing.testing.estimators import LinearRegressionEstimator
 
 class TestCausalTestOutcome(unittest.TestCase):
     """Test the TestCausalTestOutcome basic methods."""
-
-    def test_empty_adjustment_set(self):
-        ctr = CausalTestResult(
+    def setUp(self) -> None:
+        self.estimator = LinearRegressionEstimator(
             treatment="A",
             outcome="A",
             treatment_value=1,
             control_value=0,
             adjustment_set={},
+        )
+    def test_empty_adjustment_set(self):
+        ctr = CausalTestResult(
+            estimator=self.estimator,
             test_value=0,
             confidence_intervals=None,
             effect_modifier_configuration=None,
@@ -24,11 +27,7 @@ def test_empty_adjustment_set(self):
     def test_exactValue_pass(self):
         test_value = TestValue(type="ate", value=5.05)
         ctr = CausalTestResult(
-            treatment="A",
-            outcome="A",
-            treatment_value=1,
-            control_value=0,
-            adjustment_set={},
+            estimator=self.estimator,
             test_value=test_value,
             confidence_intervals=None,
             effect_modifier_configuration=None,
@@ -39,11 +38,7 @@ def test_exactValue_pass(self):
     def test_exactValue_fail(self):
         test_value = TestValue(type="ate", value=0)
         ctr = CausalTestResult(
-            treatment="A",
-            outcome="A",
-            treatment_value=1,
-            control_value=0,
-            adjustment_set={},
+            estimator=self.estimator,
             test_value=test_value,
             confidence_intervals=None,
             effect_modifier_configuration=None,
@@ -54,11 +49,7 @@ def test_exactValue_fail(self):
     def test_someEffect_pass(self):
         test_value = TestValue(type="ate", value=5.05)
         ctr = CausalTestResult(
-            treatment="A",
-            outcome="A",
-            treatment_value=1,
-            control_value=0,
-            adjustment_set={},
+            estimator=self.estimator,
             test_value=test_value,
             confidence_intervals=[4.8, 6.7],
             effect_modifier_configuration=None,
@@ -69,11 +60,7 @@ def test_someEffect_pass(self):
     def test_someEffect_fail(self):
         test_value = TestValue(type="ate", value=0)
         ctr = CausalTestResult(
-            treatment="A",
-            outcome="A",
-            treatment_value=1,
-            control_value=0,
-            adjustment_set={},
+            estimator=self.estimator,
             test_value=test_value,
             confidence_intervals=[-0.1, 0.2],
             effect_modifier_configuration=None,