Added boolean column for coverage

Author: jmafoster1

tests/discovery_tests/test_abstract_discovery.py

@@ -178,6 +178,7 @@ def test_write_dot_invalid_independence_outcome(self):
 
     def test_evaluate_tests_inestimable(self):
         scarf_df = pd.read_csv("tests/resources/data/scarf_data.csv")
+        scarf_df["completed"] = scarf_df["completed"].astype(bool)
         scarf_df = scarf_df.loc[(scarf_df["completed"]) & (scarf_df["color"] != "grey")]
 
         dag = CausalDAG()
@@ -186,6 +187,7 @@ def test_evaluate_tests_inestimable(self):
 
         abstract_discovery = AbstractDiscovery(scarf_df)
         test_results = abstract_discovery.evaluate_tests(dag)
+        print(test_results)
         expected_results = pd.DataFrame(
             [
                 {
@@ -221,7 +223,7 @@ def test_evaluate_tests_inestimable(self):
                     "expected_effect": "SomeEffect",
                     "treatment": "length_in",
                     "outcome": "completed",
-                    "effect": "negative",
+                    "effect": "positive",
                 },
                 {
                     "result": TestResult.PASS,
@@ -245,7 +247,7 @@ def test_evaluate_tests_inestimable(self):
                     "effect": "negative",
                 },
                 {
-                    "result": TestResult.PASS,
+                    "result": TestResult.INESTIMABLE,
                     "expected_effect": "NoEffect",
                     "treatment": "color",
                     "outcome": "completed",
@@ -260,6 +262,7 @@ def test_evaluate_tests_inestimable(self):
                 },
             ]
         )
+        print(expected_results)
         pd.testing.assert_frame_equal(test_results, expected_results)
 
     def test_evaluate_tests(self):