test __main__

Author: jmafoster1

causal_testing/__main__.py

@@ -17,31 +17,26 @@ def main() -> None:
     # Setup logging
     setup_logging(args.verbose)
 
-    try:
-        # Create paths object
-        paths = CausalTestingPaths(
-            dag_path=args.dag_path,
-            data_paths=args.data_paths,
-            test_config_path=args.test_config,
-            output_path=args.output,
-        )
-
-        # Create and setup framework
-        framework = CausalTestingFramework(paths, ignore_cycles=args.ignore_cycles, query=args.query)
-        framework.setup()
-
-        # Load and run tests
-        framework.load_tests()
-        results = framework.run_tests(silent=args.silent)
-
-        # Save results
-        framework.save_results(results)
-
-        logging.info("Causal testing completed successfully.")
-
-    except Exception as e:
-        logging.error("Error during causal testing: %s", str(e))
-        raise
+    # Create paths object
+    paths = CausalTestingPaths(
+        dag_path=args.dag_path,
+        data_paths=args.data_paths,
+        test_config_path=args.test_config,
+        output_path=args.output,
+    )
+
+    # Create and setup framework
+    framework = CausalTestingFramework(paths, ignore_cycles=args.ignore_cycles, query=args.query)
+    framework.setup()
+
+    # Load and run tests
+    framework.load_tests()
+    results = framework.run_tests(silent=args.silent)
+
+    # Save results
+    framework.save_results(results)
+
+    logging.info("Causal testing completed successfully.")
 
 
 if __name__ == "__main__":

tests/main_tests/test_main.py

@@ -4,6 +4,8 @@
 import pandas as pd
 from pathlib import Path
 from causal_testing.main import CausalTestingPaths, CausalTestingFramework, parse_args
+from causal_testing.__main__ import main
+from unittest.mock import patch
 
 
 class TestCausalTestingPaths(unittest.TestCase):
@@ -128,6 +130,24 @@ def test_ctf(self):
 
         self.assertEqual(tests_passed, [True])
 
+    def test_parse_args(self):
+        with unittest.mock.patch(
+            "sys.argv",
+            [
+                "causal_testing",
+                "--dag_path",
+                str(self.dag_path),
+                "--data_paths",
+                str(self.data_paths[0]),
+                "--test_config",
+                str(self.test_config_path),
+                "--output",
+                str(self.output_path.parent / "main.json"),
+            ],
+        ):
+            main()
+            self.assertTrue((self.output_path.parent / "main.json").exists())
+
     def tearDown(self):
         if self.output_path.parent.exists():
             shutil.rmtree(self.output_path.parent)