unit tests

Author: rsomers1998

tests/testing_tests/test_causal_test_outcome.py

@@ -2,6 +2,7 @@
 from causal_testing.testing.causal_test_outcome import ExactValue, SomeEffect, Positive, Negative
 from causal_testing.testing.causal_test_result import CausalTestResult, TestValue
 from causal_testing.testing.estimators import LinearRegressionEstimator
+from causal_testing.testing.validation import CausalValidator
 
 
 class TestCausalTestOutcome(unittest.TestCase):
@@ -176,3 +177,33 @@ def test_someEffect_fail(self):
                 "ci_high": 0.2,
             },
         )
+
+    def test_positive_risk_ratio_e_value(self):
+        test_value = TestValue("risk_ratio", 1.5)
+        ctr = CausalTestResult(
+            estimator=self.estimator,
+            test_value=test_value,
+            confidence_intervals=[1.2, 1.8],
+            effect_modifier_configuration=None,
+        )
+
+        cv = CausalValidator()
+        e_value, e_confidence_intervals = cv.estimate_e_value(ctr.test_value.value, ctr.confidence_intervals)
+        self.assertEqual(round(e_value, 4), 2.366)
+        self.assertEqual(round(e_confidence_intervals[0], 4), 1.6899)
+        self.assertEqual(e_confidence_intervals[1], 1)
+
+    def test_negative_risk_ratio_e_value(self):
+        test_value = TestValue("risk_ratio", 0.8)
+        ctr = CausalTestResult(
+            estimator=self.estimator,
+            test_value=test_value,
+            confidence_intervals=[0.2, 0.9],
+            effect_modifier_configuration=None,
+        )
+
+        cv = CausalValidator()
+        e_value, e_confidence_intervals = cv.estimate_e_value(ctr.test_value.value, ctr.confidence_intervals)
+        self.assertEqual(round(e_value, 4), 1.809)
+        self.assertEqual(e_confidence_intervals[0], 1)
+        self.assertEqual(round(e_confidence_intervals[1], 4), 1.4625)

tests/testing_tests/test_estimators.py

@@ -9,6 +9,7 @@
     InstrumentalVariableEstimator,
 )
 from causal_testing.specification.variable import Input
+from causal_testing.testing.validation import CausalValidator
 
 
 def plot_results_df(df):
@@ -372,6 +373,15 @@ def test_program_15_no_interaction_ate_calculated(self):
         self.assertEqual(round(ate, 1), 3.5)
         self.assertEqual([round(ci_low, 1), round(ci_high, 1)], [1.9, 5])
 
+    def test_program_11_2_with_robustness_validation(self):
+        """Test whether our linear regression estimator, as used in test_program_11_2 can correctly estimate robustness."""
+        df = self.chapter_11_df.copy()
+        linear_regression_estimator = LinearRegressionEstimator("treatments", 100, 90, set(), "outcomes", df)
+        model = linear_regression_estimator._run_linear_regression()
+
+        cv = CausalValidator()
+        self.assertEqual(round(cv.estimate_robustness(model)["treatments"], 4), 0.7353)
+
 
 class TestCausalForestEstimator(unittest.TestCase):
     """Test the linear regression estimator against the programming exercises in Section 2 of Hernán and Robins [1].