convert prints to logger

Author: christopher-wild

causal_testing/json_front/json_class.py

@@ -1,3 +1,4 @@
+import logging
 from pathlib import Path
 
 from abc import ABC
@@ -16,6 +17,8 @@
 from causal_testing.testing.causal_test_case import CausalTestCase
 from causal_testing.testing.estimators import Estimator
 
+logger = logging.getLogger(__name__)
+
 
 class JsonUtility(ABC):
     """
@@ -113,16 +116,16 @@ def execute_tests(self, effects: dict, mutates: dict, estimators: dict, f_flag:
             )
 
             concrete_tests, dummy = abstract_test.generate_concrete_tests(5, 0.05)
-            print(abstract_test)
-            print([(v.name, v.distribution) for v in abstract_test.treatment_variables])
-            print(len(concrete_tests))
+            logger.info(abstract_test)
+            logger.info([(v.name, v.distribution) for v in abstract_test.treatment_variables])
+            logger.info(len(concrete_tests))
             for concrete_test in concrete_tests:
                 executed_tests += 1
                 failed = self._execute_test_case(concrete_test, estimators[test['estimator']], f_flag)
                 if failed:
                     failures += 1
 
-        print(f"{failures}/{executed_tests} failed")
+        logger.info(f"{failures}/{executed_tests} failed")
 
     def _json_parse(self):
         """Parse a JSON input file into inputs, outputs, metas and a test plan
@@ -147,7 +150,7 @@ def _populate_metas(self):
             f.fit()
             (dist, params) = list(f.get_best(method="sumsquare_error").items())[0]
             var.distribution = getattr(scipy.stats, dist)(**params)
-            print(var.name, f"{dist}({params})")
+            logger.info(var.name + f"{dist}({params})")
 
     def _execute_test_case(self, causal_test_case: CausalTestCase, estimator: Estimator, f_flag: bool) -> bool:
         """ Executes a singular test case, prints the results and returns the test case result
@@ -174,7 +177,8 @@ def _execute_test_case(self, causal_test_case: CausalTestCase, estimator: Estima
                                 f"got {result_string}"
         if not test_passes:
             failed = True
-            print(f"    FAILED - expected {causal_test_case.expected_causal_effect}, got {causal_test_result.ate}")
+            logger.warning(
+                f"    FAILED - expected {causal_test_case.expected_causal_effect}, got {causal_test_result.ate}")
         return failed
 
     def _setup_test(self, causal_test_case: CausalTestCase, estimator: Estimator) -> tuple[CausalTestEngine, Estimator]: