add type hints for causal_test_suite.py + black

Author: christopher-wild

causal_testing/generation/abstract_causal_test_case.py

@@ -146,7 +146,6 @@ def _generate_concrete_tests(
                         + f"{constraints}\nUsing value {v.cast(model[v.z3])} instead in test\n{concrete_test}"
                     )
 
-
             if not any([vars(t) == vars(concrete_test) for t in concrete_tests]):
                 concrete_tests.append(concrete_test)
                 # Control run
@@ -162,7 +161,6 @@ def _generate_concrete_tests(
                     treatment_run["bin"] = index
                     runs.append(treatment_run)
 
-
         return concrete_tests, pd.DataFrame(runs, columns=run_columns + ["bin"])
 
     def generate_concrete_tests(

causal_testing/testing/causal_test_suite.py

@@ -1,4 +1,8 @@
 from collections import UserDict
+from typing import Type, Iterable
+from causal_testing.testing.base_test_case import BaseTestCase
+from causal_testing.testing.causal_test_case import CausalTestCase
+from causal_testing.testing.estimators import Estimator
 
 
 class CausalTestSuite(UserDict):
@@ -13,7 +17,13 @@ class CausalTestSuite(UserDict):
     base_test_case's and execute each causal_test_case with each iterator.
     """
 
-    def add_test_object(self, base_test_case, causal_test_case_list, estimators_classes, estimate_type: str = "ate"):
+    def add_test_object(
+        self,
+        base_test_case: BaseTestCase,
+        causal_test_case_list: Iterable[CausalTestCase],
+        estimators_classes: Iterable[Type[Estimator]],
+        estimate_type: str = "ate",
+    ):
         """
         A setter object to allow for the easy construction of the dictionary test suite structure
 

causal_testing/testing/estimators.py

@@ -282,7 +282,7 @@ class LinearRegressionEstimator(Estimator):
 
     def __init__(
         self,
-        treatment: tuple,    
+        treatment: tuple,
         treatment_value: float,
         control_value: float,
         adjustment_set: set,