argparse test

Author: christopher-wild

causal_testing/json_front/json_class.py

@@ -67,7 +67,6 @@ def set_path(self, json_path: str, dag_path: str, data_path: str):
         self.dag_path = Path(dag_path)
         self.data_path = Path(data_path)
 
-
     def set_variables(self, inputs: dict, outputs: dict, metas: dict):
 
         """Populate the Causal Variables
@@ -79,7 +78,6 @@ def set_variables(self, inputs: dict, outputs: dict, metas: dict):
         self.outputs = [Output(i["name"], i["type"]) for i in outputs]
         self.metas = [Meta(i["name"], i["type"], i["populate"]) for i in metas] if metas else []
 
-
     def setup(self):
         """Function to populate all the necessary parts of the json_class needed to execute tests"""
         self.modelling_scenario = Scenario(self.inputs + self.outputs + self.metas, None)
@@ -236,7 +234,7 @@ def setup_logger(log_path: str):
         setup_log.addHandler(file_handler)
 
     @staticmethod
-    def get_args() -> argparse.Namespace:
+    def get_args(test_args=None) -> argparse.Namespace:
         """Command-line arguments
 
         :return: parsed command line arguments
@@ -259,7 +257,6 @@ def get_args() -> argparse.Namespace:
             help="Specify path to file containing runtime data",
             required=True,
         )
-        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",
@@ -270,4 +267,4 @@ def get_args() -> argparse.Namespace:
             help="Specify path to file containing JSON tests, normally a .json file",
             required=True,
         )
-        return parser.parse_args()
+        return parser.parse_args(test_args)

causal_testing/testing/estimators.py

@@ -309,7 +309,6 @@ def estimate_unit_ate(self) -> float:
 
         return unit_effect * self.treatment_values - unit_effect * self.control_values, [ci_low, ci_high]
 
-
     def estimate_ate(self) -> tuple[float, list[float, float], float]:
         """Estimate the average treatment effect of the treatment on the outcome. That is, the change in outcome caused
         by changing the treatment variable from the control value to the treatment value.
@@ -531,7 +530,6 @@ def estimate_cates(self) -> pd.DataFrame:
         # Obtain CATES and confidence intervals
         conditional_ates = model.effect(effect_modifier_df, T0=self.control_values, T1=self.treatment_values).flatten()
         [ci_low, ci_high] = model.effect_interval(
-
             effect_modifier_df, T0=self.control_values, T1=self.treatment_values, alpha=0.05
         )
 

tests/json_front_tests/test_json_class.py

@@ -53,9 +53,14 @@ def test_set_metas(self):
         self.assertEqual(ctf_meta[0].name, self.json_class.metas[0].name)
         self.assertEqual(ctf_meta[0].datatype, self.json_class.metas[0].datatype)
 
+    def test_argparse(self):
+        args = self.json_class.get_args(["--data_path=data.csv", "--dag_path=dag.dot", "--json_path=tests.json"])
+        self.assertTrue(args.data_path)
+        self.assertTrue(args.dag_path)
+        self.assertTrue(args.json_path)
+
     def tearDown(self) -> None:
         remove_temp_dir_if_existent()
 
-
 def populate_example():
     pass