Fixed the tests

Author: jmafoster1

tests/discovery_tests/test_abstract_discovery.py

@@ -195,21 +195,21 @@ def test_evaluate_tests(self):
                     "effect": "positive",
                 },
                 {
-                    "result": TestResult.INESTIMABLE,
+                    "result": TestResult.PASS,
                     "expected_effect": "NoEffect",
                     "treatment": "large_gauge",
                     "outcome": "length_in",
-                    "effect": None,
+                    "effect": "positive",
                 },
                 {
-                    "result": TestResult.INESTIMABLE,
+                    "result": TestResult.PASS,
                     "expected_effect": "NoEffect",
                     "treatment": "length_in",
                     "outcome": "color",
                     "effect": None,
                 },
                 {
-                    "result": TestResult.INESTIMABLE,
+                    "result": TestResult.PASS,
                     "expected_effect": "NoEffect",
                     "treatment": "color",
                     "outcome": "length_in",
@@ -223,7 +223,7 @@ def test_evaluate_tests(self):
                     "effect": "negative",
                 },
                 {
-                    "result": TestResult.INESTIMABLE,
+                    "result": TestResult.PASS,
                     "expected_effect": "NoEffect",
                     "treatment": "large_gauge",
                     "outcome": "color",
@@ -251,7 +251,7 @@ def test_evaluate_tests(self):
                     "effect": None,
                 },
                 {
-                    "result": TestResult.INESTIMABLE,
+                    "result": TestResult.PASS,
                     "expected_effect": "NoEffect",
                     "treatment": "completed",
                     "outcome": "color",

tests/discovery_tests/test_hill_climber_discovery.py

@@ -106,10 +106,8 @@ def test_evaluate_fitness(self):
 
         hill_climber = HillClimberDiscovery(scarf_df)
         fitness_values, problem_edges = hill_climber.evaluate_fitness(dag)
-        expected_fitness_values = (1.5, -2, -2.5)
+        expected_fitness_values = (4, -2, 0)
         expected_problem_edges = [
-            ("length_in", "color"),
-            ("color", "length_in"),
             ("length_in", "completed"),
             ("large_gauge", "completed"),
         ]

tests/discovery_tests/test_nsga_discovery.py

@@ -30,11 +30,10 @@ def test_multiobjective_fitness(self):
         dag.add_edges_from([("length_in", "completed"), ("large_gauge", "completed")])
         nsga = NSGADiscovery(scarf_df)
 
+        expected = np.array([2, 2, 2, 0.0, 0.0, 2])
+        multiobjective_fitness = nsga.multi_objective_fitness(None, nsga.causal_dag_to_binary_string(dag), None)
         self.assertTrue(
-            np.array_equal(
-                nsga.multi_objective_fitness(None, nsga.causal_dag_to_binary_string(dag), None),
-                np.array([1.5, 1.5, 1.0, 0.0, 0.0, 1.5]),
-            )
+            np.array_equal(multiobjective_fitness, expected), f"Arrays differ: {expected} != {multiobjective_fitness}"
         )
 
     def test_discovery_edges(self):

tests/main_tests/test_main.py

@@ -148,7 +148,7 @@ def test_create_test_case_invalid_estimator(self):
         with self.assertRaises(ValueError) as e:
             framework.create_causal_test({"estimator": "InvalidEstimator"}, None)
         self.assertEqual(
-            f"Unsupported estimator InvalidEstimator. Supported: ['CubicSplineEstimator', 'IPCWEstimator', 'InstrumentalVariableEstimator', 'LinearRegressionEstimator', 'LogisticRegressionEstimator']. "
+            f"Unsupported estimator InvalidEstimator. Supported: ['CubicSplineEstimator', 'IPCWEstimator', 'InstrumentalVariableEstimator', 'LinearRegressionEstimator', 'LogisticRegressionEstimator', 'MultinomialRegressionEstimator']. "
             "If you have implemented a custom estimator, you will need to add this to your entrypoints via your "
             "pyproject.toml file.",
             str(e.exception),