Check formula is used only for Regression Estimator

Author: christopher-wild

causal_testing/json_front/json_class.py

@@ -22,7 +22,7 @@
 from causal_testing.specification.variable import Input, Meta, Output
 from causal_testing.testing.causal_test_case import CausalTestCase
 from causal_testing.testing.causal_test_result import CausalTestResult
-from causal_testing.testing.estimators import Estimator
+from causal_testing.testing.estimators import Estimator, LinearRegressionEstimator
 from causal_testing.testing.base_test_case import BaseTestCase
 from causal_testing.testing.causal_test_adequacy import DataAdequacy
 
@@ -165,12 +165,12 @@ def _run_coefficient_test(self, test: dict, f_flag: bool, effects: dict):
         )
         failed, result = self._execute_test_case(causal_test_case=causal_test_case, test=test, f_flag=f_flag)
         msg = (
-                f"Executing test: {test['name']} \n"
-                + f"  {causal_test_case} \n"
-                + "  "
-                + ("\n  ").join(str(result).split("\n"))
-                + "==============\n"
-                + f"  Result: {'FAILED' if failed else 'Passed'}"
+            f"Executing test: {test['name']} \n"
+            + f"  {causal_test_case} \n"
+            + "  "
+            + ("\n  ").join(str(result).split("\n"))
+            + "==============\n"
+            + f"  Result: {'FAILED' if failed else 'Passed'}"
         )
         self._append_to_file(msg, logging.INFO)
         return failed, result
@@ -192,11 +192,11 @@ def _run_concrete_metamorphic_test(self, test: dict, f_flag: bool, effects: dict
         failed, msg = self._execute_test_case(causal_test_case=causal_test_case, test=test, f_flag=f_flag)
 
         msg = (
-                f"Executing concrete test: {test['name']} \n"
-                + f"treatment variable: {test['treatment_variable']} \n"
-                + f"outcome_variable = {outcome_variable} \n"
-                + f"control value = {test['control_value']}, treatment value = {test['treatment_value']} \n"
-                + f"Result: {'FAILED' if failed else 'Passed'}"
+            f"Executing concrete test: {test['name']} \n"
+            + f"treatment variable: {test['treatment_variable']} \n"
+            + f"outcome_variable = {outcome_variable} \n"
+            + f"control value = {test['control_value']}, treatment value = {test['treatment_value']} \n"
+            + f"Result: {'FAILED' if failed else 'Passed'}"
         )
         self._append_to_file(msg, logging.INFO)
         return failed, msg
@@ -225,13 +225,13 @@ def _run_metamorphic_tests(self, test: dict, f_flag: bool, effects: dict, mutate
         failures, _ = self._execute_tests(concrete_tests, test, f_flag)
 
         msg = (
-                f"Executing test: {test['name']} \n"
-                + "  abstract_test \n"
-                + f"  {abstract_test} \n"
-                + f"  {abstract_test.treatment_variable.name},"
-                + f"  {abstract_test.treatment_variable.distribution} \n"
-                + f"  Number of concrete tests for test case: {str(len(concrete_tests))} \n"
-                + f"  {failures}/{len(concrete_tests)} failed for {test['name']}"
+            f"Executing test: {test['name']} \n"
+            + "  abstract_test \n"
+            + f"  {abstract_test} \n"
+            + f"  {abstract_test.treatment_variable.name},"
+            + f"  {abstract_test.treatment_variable.distribution} \n"
+            + f"  Number of concrete tests for test case: {str(len(concrete_tests))} \n"
+            + f"  {failures}/{len(concrete_tests)} failed for {test['name']}"
         )
         self._append_to_file(msg, logging.INFO)
         return failures, msg
@@ -257,7 +257,7 @@ def _populate_metas(self):
             meta.populate(self.data_collector.data)
 
     def _execute_test_case(
-            self, causal_test_case: CausalTestCase, test: Mapping, f_flag: bool
+        self, causal_test_case: CausalTestCase, test: Mapping, f_flag: bool
     ) -> (bool, CausalTestResult):
         """Executes a singular test case, prints the results and returns the test case result
         :param causal_test_case: The concrete test case to be executed
@@ -309,11 +309,14 @@ def _setup_test(self, causal_test_case: CausalTestCase, test: Mapping) -> Estima
         """
         estimator_kwargs = {}
         if "formula" in test:
+            if test["estimator"] != LinearRegressionEstimator:
+                raise TypeError("Currently only LinearRegressionEstimator supports the use of formulas")
             estimator_kwargs["formula"] = test["formula"]
             estimator_kwargs["adjustment_set"] = {}
         else:
             minimal_adjustment_set = self.causal_specification.causal_dag.identification(
-                causal_test_case.base_test_case)
+                causal_test_case.base_test_case
+            )
             minimal_adjustment_set = minimal_adjustment_set - {causal_test_case.treatment_variable}
             estimator_kwargs["adjustment_set"] = minimal_adjustment_set
 
@@ -377,7 +380,7 @@ def get_args(test_args=None) -> argparse.Namespace:
         parser.add_argument(
             "-w",
             help="Specify to overwrite any existing output files. This can lead to the loss of existing outputs if not "
-                 "careful",
+            "careful",
             action="store_true",
         )
         parser.add_argument(