Add new test files

Author: christopher-wild

tests/json_front_tests/test_json_class.py

@@ -99,7 +99,7 @@ def test_f_flag(self):
         effects = {"NoEffect": NoEffect()}
         mutates = {
             "Increase": lambda x: self.json_class.scenario.treatment_variables[x].z3
-                                  > self.json_class.scenario.variables[x].z3
+            > self.json_class.scenario.variables[x].z3
         }
         estimators = {"LinearRegressionEstimator": LinearRegressionEstimator}
         with self.assertRaises(StatisticsError):
@@ -149,7 +149,7 @@ def test_run_json_tests_from_json(self):
         effects = {"NoEffect": NoEffect()}
         mutates = {
             "Increase": lambda x: self.json_class.scenario.treatment_variables[x].z3
-                                  > self.json_class.scenario.variables[x].z3
+            > self.json_class.scenario.variables[x].z3
         }
         estimators = {"LinearRegressionEstimator": LinearRegressionEstimator}
 
@@ -176,7 +176,7 @@ def test_generate_tests_from_json_no_dist(self):
         effects = {"NoEffect": NoEffect()}
         mutates = {
             "Increase": lambda x: self.json_class.scenario.treatment_variables[x].z3
-                                  > self.json_class.scenario.variables[x].z3
+            > self.json_class.scenario.variables[x].z3
         }
         estimators = {"LinearRegressionEstimator": LinearRegressionEstimator}
 
@@ -206,7 +206,7 @@ def test_formula_in_json_test(self):
         effects = {"Positive": Positive()}
         mutates = {
             "Increase": lambda x: self.json_class.scenario.treatment_variables[x].z3
-                                  > self.json_class.scenario.variables[x].z3
+            > self.json_class.scenario.variables[x].z3
         }
         estimators = {"LinearRegressionEstimator": LinearRegressionEstimator}
 
@@ -259,7 +259,7 @@ def test_concrete_generate_params(self):
         effects = {"NoEffect": NoEffect()}
         mutates = {
             "Increase": lambda x: self.json_class.scenario.treatment_variables[x].z3
-                                  > self.json_class.scenario.variables[x].z3
+            > self.json_class.scenario.variables[x].z3
         }
         estimators = {"LinearRegressionEstimator": LinearRegressionEstimator}
 
@@ -294,16 +294,15 @@ def test_estimator_formula_type_check(self):
             ]
         }
         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 = {"CausalForestEstimator": CausalForestEstimator}
         with self.assertRaises(TypeError):
             self.json_class.run_json_tests(effects=effects, mutates=mutates, estimators=estimators, f_flag=False)
 
-    effects = {"Positive": Positive()}
-    mutates = {
-        "Increase": lambda x: self.json_class.scenario.treatment_variables[x].z3
-                              > self.json_class.scenario.variables[x].z3
-    }
-    estimators = {"CausalForestEstimator": CausalForestEstimator}
 
     def test_constructive_back_door_not_met(self):
         example_test = {
@@ -334,7 +333,6 @@ def test_constructive_back_door_not_met(self):
         test_data_dir_path = Path("tests/resources/data")
         dag_path = str(test_data_dir_path / "dag_not_descendent.dot")
         data_path = [str(test_data_dir_path / "not_descendent.csv")]
-
         input_dict_list = [
             {"name": "X", "datatype": float},
             {"name": "Z", "datatype": float},

tests/resources/data/dag_not_descendent.dot

@@ -0,0 +1 @@
+digraph G {X -> Y; Z -> X; Z -> Y}

tests/resources/data/not_descendent.csv

@@ -0,0 +1,2 @@
+X,Y,Z
+0,0,0