add command line args + codacy

Author: christopher-wild

causal_testing/json_front/json_class.py

@@ -232,4 +232,13 @@ def get_args() -> argparse.Namespace:
             "-f", help="if included, the script will stop if a test fails", action="store_true")
         parser.add_argument(
             "--log_path", help="Specify a directory to change the location of the log file", default=".")
+        parser.add_argument(
+            "--data_path", help="Specify path to file containing runtime data", required=True
+        )
+        parser.add_argument(
+            "--dag_path", help="Specify path to file containing the DAG, normally a .dot file", required=True
+        )
+        parser.add_argument(
+            "--json_path", help="Specify path to file containing JSON tests, normally a .json file", required=True
+        )
         return parser.parse_args()

examples/poisson/run_causal_tests.py

@@ -1,4 +1,4 @@
-
+import logging
 import numpy as np
 import pandas as pd
 import scipy
@@ -10,10 +10,6 @@
 from causal_testing.testing.estimators import Estimator
 from causal_testing.specification.scenario import Scenario
 from causal_testing.specification.variable import Input, Output, Meta
-import logging
-data_path = "data.csv"
-dag_path = "dag.dot"
-json_path = "causal_tests.json"
 
 
 class WidthHeightEstimator(LinearRegressionEstimator):
@@ -147,10 +143,9 @@ def logger_setup():
     "LinearRegressionEstimator": LinearRegressionEstimator,
 }
 
-
 # Create input structure required to create a modelling scenario
-modelling_inputs = [Input(i['name'], i['type'], distributions[i['distribution']]) for i in inputs] +\
-                   [Output(i['name'], i['type']) for i in outputs] +\
+modelling_inputs = [Input(i['name'], i['type'], distributions[i['distribution']]) for i in inputs] + \
+                   [Output(i['name'], i['type']) for i in outputs] + \
                    [Meta(i['name'], i['type'], populates[i['populate']]) for i in metas] if metas else list()
 
 # Create modelling scenario to access z3 variable mirrors
@@ -181,9 +176,9 @@ def add_modelling_assumptions(self, estimation_model: Estimator):
 
 if __name__ == "__main__":
     args = MyJsonUtility.get_args()
-
     json_utility = MyJsonUtility(args.log_path)  # Create an instance of the extended JsonUtility class
-    json_utility.set_path(json_path, dag_path, data_path)  # Set the path to the data.csv, dag.dot and causal_tests.json file
+    json_utility.set_path(args.json_path, args.dag_path,
+                          args.data_path)  # Set the path to the data.csv, dag.dot and causal_tests.json file
 
     # Load the Causal Variables into the JsonUtility class ready to be used in the tests
     json_utility.set_variables(inputs, outputs, metas, distributions, populates)