Add unittest for formula

Author: christopher-wild

causal_testing/json_front/json_class.py

@@ -120,6 +120,8 @@ def generate_tests(self, effects: dict, mutates: dict, estimators: dict, f_flag:
     def _execute_tests(self, concrete_tests, estimators, test, f_flag):
         failures = 0
         test["estimator"] = estimators[test["estimator"]]
+        if "formula" in test:
+            self._append_to_file(f"Estimator formula used for test: {test['formula']}")
         for concrete_test in concrete_tests:
             failed = self._execute_test_case(concrete_test, test, f_flag)
             if failed:
@@ -234,7 +236,7 @@ def _append_to_file(self, line: str, log_level: int = None):
         """
         with open(self.output_path, "a", encoding="utf-8") as f:
             f.write(
-                line + "\n",
+                line
             )
         if log_level:
             logger.log(level=log_level, msg=line)

tests/json_front_tests/test_json_class.py

@@ -6,7 +6,7 @@
 import json
 
 from causal_testing.testing.estimators import LinearRegressionEstimator
-from causal_testing.testing.causal_test_outcome import NoEffect
+from causal_testing.testing.causal_test_outcome import NoEffect, Positive
 from tests.test_helpers import create_temp_dir_if_non_existent, remove_temp_dir_if_existent
 from causal_testing.json_front.json_class import JsonUtility, CausalVariables
 from causal_testing.specification.variable import Input, Output, Meta
@@ -127,6 +127,34 @@ def test_generate_tests_from_json(self):
             temp_out = reader.readlines()
         self.assertIn("failed", temp_out[-1])
 
+    def test_formula_in_json_test(self):
+        example_test = {
+            "tests": [
+                {
+                    "name": "test1",
+                    "mutations": {"test_input": "Increase"},
+                    "estimator": "LinearRegressionEstimator",
+                    "estimate_type": "ate",
+                    "effect_modifiers": [],
+                    "expectedEffect": {"test_output": "Positive"},
+                    "skip": False,
+                    "formula": "test_output ~ test_input"
+                }
+            ]
+        }
+        self.json_class.test_plan = example_test
+        effects = {"Positive": Positive()}
+        mutates = {
+            "Increase": lambda x: self.json_class.scenario.treatment_variables[x].z3
+                                  > self.json_class.scenario.variables[x].z3
+        }
+        estimators = {"LinearRegressionEstimator": LinearRegressionEstimator}
+
+        self.json_class.generate_tests(effects, mutates, estimators, False)
+        with open("temp_out.txt", 'r') as reader:
+            temp_out = reader.readlines()
+        self.assertIn("test_output ~ test_input", ''.join(temp_out))
+
     def tearDown(self) -> None:
         pass
         # remove_temp_dir_if_existent()