Black and pylint

Author: jmafoster1

causal_testing/json_front/json_class.py

@@ -225,7 +225,9 @@ def _execute_test_case(self, causal_test_case: CausalTestCase, test: Iterable[Ma
             logger.warning("   FAILED- expected %s, got %s", causal_test_case.expected_causal_effect, result_string)
         return failed
 
-    def _setup_test(self, causal_test_case: CausalTestCase, test: Mapping, conditions: list[str]=None) -> tuple[CausalTestEngine, Estimator]:
+    def _setup_test(
+        self, causal_test_case: CausalTestCase, test: Mapping, conditions: list[str] = None
+    ) -> tuple[CausalTestEngine, Estimator]:
         """Create the necessary inputs for a single test case
         :param causal_test_case: The concrete test case to be executed
         :param test: Single JSON test definition stored in a mapping (dict)

causal_testing/specification/causal_dag.py

@@ -521,7 +521,10 @@ def identification(self, base_test_case: BaseTestCase):
         minimal_adjustment_set = min(minimal_adjustment_sets, key=len)
         return minimal_adjustment_set
 
-    def to_dot(self):
+    def to_dot(self) -> str:
+        """Return a string of the DOT representation of the causal DAG.
+        :return DOT string of the DAG.
+        """
         return str(nx.nx_pydot.to_pydot(self.graph))
 
     def __str__(self):

tests/json_front_tests/test_json_class.py

@@ -197,22 +197,22 @@ def test_formula_in_json_test(self):
                     "effect_modifiers": [],
                     "expectedEffect": {"test_output": "Positive"},
                     "skip": False,
-                    "formula": "test_output ~ test_input"
+                    "formula": "test_output ~ test_input",
                 }
             ]
         }
         self.json_class.test_plan = example_test
         effects = {"Positive": Positive()}
         mutates = {
             "Increase": lambda x: self.json_class.scenario.treatment_variables[x].z3
-                                  > self.json_class.scenario.variables[x].z3
+            > self.json_class.scenario.variables[x].z3
         }
         estimators = {"LinearRegressionEstimator": LinearRegressionEstimator}
 
         self.json_class.generate_tests(effects, mutates, estimators, False)
-        with open("temp_out.txt", 'r') as reader:
+        with open("temp_out.txt", "r") as reader:
             temp_out = reader.readlines()
-        self.assertIn("test_output ~ test_input", ''.join(temp_out))
+        self.assertIn("test_output ~ test_input", "".join(temp_out))
 
     def tearDown(self) -> None:
         remove_temp_dir_if_existent()