Added extra test for coverage

Author: jmafoster1

tests/discovery_tests/test_abstract_discovery.py

@@ -176,6 +176,92 @@ def test_write_dot_invalid_independence_outcome(self):
         with self.assertRaises(ValueError):
             abstract_discovery.write_dot(dag, "dag.dot")
 
+    def test_evaluate_tests_inestimable(self):
+        scarf_df = pd.read_csv("tests/resources/data/scarf_data.csv")
+        scarf_df = scarf_df.loc[(scarf_df["completed"]) & (scarf_df["color"] != "grey")]
+
+        dag = CausalDAG()
+        dag.add_nodes_from(scarf_df.columns)
+        dag.add_edges_from([("length_in", "completed"), ("large_gauge", "completed")])
+
+        abstract_discovery = AbstractDiscovery(scarf_df)
+        test_results = abstract_discovery.evaluate_tests(dag)
+        expected_results = pd.DataFrame(
+            [
+                {
+                    "result": TestResult.PASS,
+                    "expected_effect": "NoEffect",
+                    "treatment": "length_in",
+                    "outcome": "large_gauge",
+                    "effect": "negative",
+                },
+                {
+                    "result": TestResult.PASS,
+                    "expected_effect": "NoEffect",
+                    "treatment": "large_gauge",
+                    "outcome": "length_in",
+                    "effect": "negative",
+                },
+                {
+                    "result": TestResult.PASS,
+                    "expected_effect": "NoEffect",
+                    "treatment": "length_in",
+                    "outcome": "color",
+                    "effect": None,
+                },
+                {
+                    "result": TestResult.PASS,
+                    "expected_effect": "NoEffect",
+                    "treatment": "color",
+                    "outcome": "length_in",
+                    "effect": None,
+                },
+                {
+                    "result": TestResult.FAIL,
+                    "expected_effect": "SomeEffect",
+                    "treatment": "length_in",
+                    "outcome": "completed",
+                    "effect": "negative",
+                },
+                {
+                    "result": TestResult.PASS,
+                    "expected_effect": "NoEffect",
+                    "treatment": "large_gauge",
+                    "outcome": "color",
+                    "effect": None,
+                },
+                {
+                    "result": TestResult.PASS,
+                    "expected_effect": "NoEffect",
+                    "treatment": "color",
+                    "outcome": "large_gauge",
+                    "effect": None,
+                },
+                {
+                    "result": TestResult.FAIL,
+                    "expected_effect": "SomeEffect",
+                    "treatment": "large_gauge",
+                    "outcome": "completed",
+                    "effect": "negative",
+                },
+                {
+                    "result": TestResult.PASS,
+                    "expected_effect": "NoEffect",
+                    "treatment": "color",
+                    "outcome": "completed",
+                    "effect": None,
+                },
+                {
+                    "result": TestResult.INESTIMABLE,
+                    "expected_effect": "NoEffect",
+                    "treatment": "completed",
+                    "outcome": "color",
+                    "effect": None,
+                },
+            ]
+        )
+        pd.testing.assert_frame_equal(test_results, expected_results)
+
     def test_evaluate_tests(self):
         scarf_df = pd.read_csv("tests/resources/data/scarf_data.csv")