Extra coverage

Author: jmafoster1

causal_testing/generation/abstract_causal_test_case.py

@@ -230,7 +230,11 @@ def generate_concrete_tests(
                 pre_break = True
                 break
             if issubclass(self.treatment_variable.datatype, Enum) and set(
-                itertools.product(self.treatment_variable.datatype, self.treatment_variable.datatype)
+                {
+                    (x, y)
+                    for x, y in itertools.product(self.treatment_variable.datatype, self.treatment_variable.datatype)
+                    if x != y
+                }
             ).issubset(zip(control_values, treatment_values)):
                 pre_break = True
                 break

tests/generation_tests/test_abstract_test_case.py

@@ -76,15 +76,15 @@ def test_generate_boolean_concrete_test_cases(self):
         abstract = AbstractCausalTestCase(
             scenario=scenario,
             intervention_constraints={
-                And(scenario.treatment_variables[self.X5.name].z3 == True, scenario.variables[self.X5.name].z3 == False)
+                scenario.treatment_variables[self.X5.name].z3 != scenario.variables[self.X5.name].z3
             },
             treatment_variable=self.X5,
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )
         concrete_tests, runs = abstract.generate_concrete_tests(2)
-        assert len(concrete_tests) == 1, "Expected 1 concrete test"
-        assert len(runs) == 1, "Expected 1 run"
+        assert len(concrete_tests) == 2, "Expected 2 concrete test"
+        assert len(runs) == 2, "Expected 2 run"
 
     def test_generate_enum_concrete_test_cases(self):
         scenario = Scenario({self.Car})
@@ -98,7 +98,7 @@ def test_generate_enum_concrete_test_cases(self):
             expected_causal_effect={self.Y: Positive()},
             effect_modifiers=None,
         )
-        concrete_tests, runs = abstract.generate_concrete_tests(2)
+        concrete_tests, runs = abstract.generate_concrete_tests(10)
         assert len(concrete_tests) == 2, "Expected 2 concrete tests"
         assert len(runs) == 2, "Expected 2 runs"