Changed it to use post_init

Author: jmafoster1

causal_testing/testing/base_test_case.py

@@ -14,9 +14,10 @@ class BaseTestCase:
     :param effect: A string representing the effect, current support effects are 'direct' and 'total'
     """
 
-    def __init__(self, treatment_variable: Variable, outcome_variable: Variable, effect: str = Effect.TOTAL.value):
-        if treatment_variable == outcome_variable:
-            raise ValueError(f"Treatment variable {treatment_variable} cannot also be the outcome.")
-        self.treatment_variable = treatment_variable
-        self.outcome_variable = outcome_variable
-        self.effect = effect
+    treatment_variable: Variable
+    outcome_variable: Variable
+    effect: str = Effect.TOTAL.value
+
+    def __post_init__(self):
+        if self.treatment_variable == self.outcome_variable:
+            raise ValueError(f"Treatment variable {self.treatment_variable} cannot also be the outcome variable.")

tests/testing_tests/test_causal_test_case.py

@@ -98,6 +98,10 @@ def setUp(self) -> None:
     def tearDown(self) -> None:
         shutil.rmtree(self.temp_dir_path)
 
+    def test_invalid_base_test_case(self):
+        with self.assertRaises(ValueError):
+            BaseTestCase(self.A, self.A)
+
     def test_check_minimum_adjustment_set(self):
         """Check that the minimum adjustment set is correctly made"""
         minimal_adjustment_set = self.causal_dag.identification(self.base_test_case_A_C)