codecov

Author: jmafoster1

causal_testing/testing/causal_test_result.py

@@ -28,16 +28,8 @@ def __init__(
         self.error_message = error_message
 
     def __str__(self):
-        def push(s, inc="  "):
-            return inc + str(s).replace("\n", "\n" + inc)
-
         result_str = str(self.effect_estimate.value.to_dict())
-        if "\n" in result_str:
-            result_str = "\n" + push(self.effect_estimate.value)
-        if isinstance(self.estimator.base_test_case.treatment_variable, list):
-            treatment = [x.name for x in self.estimator.base_test_case.treatment_variable]
-        else:
-            treatment = self.estimator.base_test_case.treatment_variable.name
+        treatment = self.estimator.base_test_case.treatment_variable.name
         base_str = (
             f"Causal Test Result\n==============\n"
             f"Treatment: {treatment}\n"

tests/main_tests/test_main.py

@@ -174,6 +174,15 @@ def test_ctf_batches(self):
 
         self.assertEqual([result["passed"] for result in all_results], [True])
 
+    def test_ctf_exception(self):
+        framework = CausalTestingFramework(self.paths, query="test_input < 0")
+        framework.setup()
+
+        # Load and run tests
+        framework.load_tests()
+        with self.assertRaises(ValueError):
+            framework.run_tests()
+
     def test_ctf_batches_exception_silent(self):
         framework = CausalTestingFramework(self.paths, query="test_input < 0")
         framework.setup()

tests/testing_tests/test_causal_effect.py

@@ -75,7 +75,7 @@ def test_Positive_ate_pass(self):
     def test_Positive_risk_ratio_pass(self):
         ctr = CausalTestResult(
             estimator=self.estimator,
-            effect_estimate=EffectEstimate(type="ate", value=pd.Series(5.05)),
+            effect_estimate=EffectEstimate(type="risk_ratio", value=pd.Series(5.05)),
         )
         ev = Positive()
         self.assertTrue(ev.apply(ctr))