Testing main.py

Author: jmafoster1

tests/main_tests/test_main.py

@@ -1,7 +1,8 @@
 import unittest
 import shutil
+import json
 from pathlib import Path
-from causal_testing.main import CausalTestingPaths
+from causal_testing.main import CausalTestingPaths, CausalTestingFramework
 
 
 class TestCausalTestingPaths(unittest.TestCase):
@@ -35,3 +36,45 @@ def test_output_file_created(self):
     def tearDown(self):
         if self.output_path.parent.exists():
             shutil.rmtree(self.output_path.parent)
+
+
+class TestCausalTestingFramework(unittest.TestCase):
+    def setUp(self):
+        self.dag_path = "tests/resources/data/dag.dot"
+        self.data_paths = ["tests/resources/data/data.csv"]
+        self.test_config_path = "tests/resources/data/tests.json"
+        self.output_path = Path("results/results.json")
+
+    def test_ctf(self):
+        # Create paths object
+        paths = CausalTestingPaths(
+            dag_path=self.dag_path,
+            data_paths=self.data_paths,
+            test_config_path=self.test_config_path,
+            output_path=self.output_path,
+        )
+
+        # Create and setup framework
+        framework = CausalTestingFramework(paths)
+        framework.setup()
+
+        # Load and run tests
+        framework.load_tests()
+        results = framework.run_tests()
+
+        # Save results
+        framework.save_results(results)
+
+        with open(self.test_config_path, "r", encoding="utf-8") as f:
+            test_configs = json.load(f)
+
+        tests_passed = [
+            test_case.expected_causal_effect.apply(result) if result.test_value.type != "Error" else False
+            for test_config, test_case, result in zip(test_configs["tests"], framework.test_cases, results)
+        ]
+
+        self.assertEqual(tests_passed, [True])
+
+    def tearDown(self):
+        if self.output_path.parent.exists():
+            shutil.rmtree(self.output_path.parent)

tests/resources/data/data.csv

@@ -1,2 +1,6 @@
-index,test_input,test_input_no_dist,test_output
-0,1.0,1.0,2.0
+test_input,test_input_no_dist,test_output,B,C
+1.0,1.1,2.2,0,0
+2.0,1.1,2.8,0,0
+3.0,1.0,1.0,0,0
+4.0,1.2,6.0,0,0
+5.0,0.9,2.5,0,0

tests/resources/data/tests.json

@@ -1,11 +1,11 @@
 {
   "tests": [{
     "name": "test1",
-    "mutations": {},
-    "estimator": null,
-    "estimate_type": null,
+    "treatment_variable": "test_input",
+    "estimator": "LinearRegressionEstimator",
+    "estimate_type": "coefficient",
     "effect_modifiers": [],
-    "expected_effect": {},
+    "expected_effect": {"test_output": "NoEffect"},
     "skip": false
   }]
 }