def test_complies_true_not_in_var_values(self, var_values: Dict[str, int],
name: str, int_set: Set[int]):
assume(name not in var_values)
constraint = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
self.assertTrue(constraint.complies(var_values=var_values))
def test_complies_true_in_var_values(self, var_values: Dict[str, int],
name: str, int_set: Set[int]):
var_values[name] = random.sample(list(int_set), 1)[0]
constraint = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
self.assertTrue(constraint.complies(var_values=var_values))
def test_merge_false(self):
self._mock_constraint_factory.reset_mock()
constraint = VarValueConstraint(
name="mock",
values={1, 2, 3},
constraint_factory=self._mock_constraint_factory)
second_constraint = create_autospec(IConstraint)
with self.assertRaises(ValueError):
constraint.merge(second_constraint)
def test_complies_false_not_in_var_values(self, var_values: Dict[str, int],
name: str, int_set: Set[int],
test_int: int):
assume(test_int not in int_set)
var_values[name] = test_int
constraint = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
self.assertFalse(constraint.complies(var_values=var_values))
def test_hash_not_equal_set(self, name: str, int_sets: List[Set[int]]):
constraint_one = VarValueConstraint(
name=name,
values=int_sets[0],
constraint_factory=self._mock_constraint_factory)
constraint_two = VarValueConstraint(
name=name,
values=int_sets[1],
constraint_factory=self._mock_constraint_factory)
self.assertNotEqual(hash(constraint_one), hash(constraint_two))
def test_hash_not_equal_name(self, names: List[str], int_set: Set[int]):
constraint_one = VarValueConstraint(
name=names[0],
values=int_set,
constraint_factory=self._mock_constraint_factory)
constraint_two = VarValueConstraint(
name=names[1],
values=int_set,
constraint_factory=self._mock_constraint_factory)
self.assertNotEqual(hash(constraint_one), hash(constraint_two))
def test_hash_equal(self, name: str, int_set: Set[int]):
constraint_one = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
constraint_two = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
self.assertEqual(hash(constraint_one), hash(constraint_two))
def test_eq_true(self, name: str, int_set: Set[int]):
constraint = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
second_constraint = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
self.assertEqual(constraint, second_constraint)
def test_is_possible(
self,
name: str,
int_set: Set[int],
):
constraint = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
self.assertEqual(len(int_set) > 0, constraint.is_possible())
def test_can_merge_constraint_false(
self,
name: str,
int_set: Set[int],
):
constraint = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
merge_constraint = create_autospec(IConstraint)
self.assertFalse(constraint.can_merge(merge_constraint))
def test_can_merge_constraint_true(self, name: str, int_set_one: Set[int],
int_set_two: Set[int]):
constraint = VarValueConstraint(
name=name,
values=int_set_one,
constraint_factory=self._mock_constraint_factory)
second_constraint = VarValueConstraint(
name=name,
values=int_set_two,
constraint_factory=self._mock_constraint_factory)
self.assertTrue(constraint.can_merge(second_constraint))
def test_eq_false_set(self, name: str, int_set_one: Set[int],
int_set_two: Set[int]):
assume(int_set_one != int_set_two)
constraint = VarValueConstraint(
name=name,
values=int_set_one,
constraint_factory=self._mock_constraint_factory)
second_constraint = VarValueConstraint(
name=name,
values=int_set_two,
constraint_factory=self._mock_constraint_factory)
self.assertNotEqual(constraint, second_constraint)
def test_repr(self, name: str, int_set: Set[int]):
constraint = VarValueConstraint(
name=name,
values=int_set,
constraint_factory=self._mock_constraint_factory)
self.assertEqual(f"VarValueConstraint: name={name}, values={int_set}",
repr(constraint))
def test_merge_constraint_some_overlap(self, name: str,
int_set_one: Set[int],
int_set_two: Set[int]):
self._mock_constraint_factory.reset_mock()
int_set_one.add(random.sample(list(int_set_two), 1)[0])
constraint = VarValueConstraint(
name=name,
values=int_set_one,
constraint_factory=self._mock_constraint_factory)
second_constraint = VarValueConstraint(
name=name,
values=int_set_two,
constraint_factory=self._mock_constraint_factory)
self.assertEqual(self._mock_constraint,
constraint.merge(second_constraint))
self._mock_constraint_factory.var_value_constraint.assert_called_once_with(
name=name, values=int_set_one.intersection(int_set_two))
def test_merge_constraint_no_overlap(
self,
name: str,
int_set: Set[int],
):
int_set_one = set(random.sample(list(int_set), len(int_set) // 2))
int_set_two = int_set - int_set_one
assume(len(int_set_one.intersection(int_set_two)) == 0)
self._mock_constraint_factory.reset_mock()
constraint = VarValueConstraint(
name=name,
values=int_set_one,
constraint_factory=self._mock_constraint_factory)
second_constraint = VarValueConstraint(
name=name,
values=int_set_two,
constraint_factory=self._mock_constraint_factory)
self.assertEqual(self._mock_constraint_factory.impossible_constraint,
constraint.merge(second_constraint))
def var_value_constraint(self, name: str,
values: Set[int]) -> IVarValueConstraint:
return VarValueConstraint(name=name,
values=values,
constraint_factory=self)