def test_constraints_with_more_than_2_neighbors_arent_added(self):
constraints = [(('A', 'B', 'C'), self.constraint)]
arcs_result = all_arcs(constraints)
arcs_expected = set()
self.assertEqual(arcs_result, arcs_expected)
def test_adds_pairs_in_both_directions(self):
constraints = [(('A', 'B'), self.constraint)]
arcs_result = all_arcs(constraints)
arcs_expected = set([('A', 'B'), ('B', 'A')])
self.assertEqual(arcs_result, arcs_expected)
def test_all_arcs(self):
# does not check that the constraint function is well created
variables = map(fst, list(all_arcs(self.constraints)))
arcs = [('A1', 'B2'), ('B2', 'A1'), ('A1', 'C2'), ('C2', 'A1'),
('D2', 'A1'), ('A1', 'D2'), ('D2', 'C1'), ('C1', 'D2'),
('E2', 'A1'), ('A1', 'E2')]
self.assertEqual(variables, arcs)
def test_constraints_with_more_than_2_neighbors_arent_added(self):
constraints = [(('A', 'B', 'C'), self.constraint)]
arcs_result = all_arcs(constraints)
arcs_expected = set()
self.assertEqual(arcs_result, arcs_expected)
def test_adds_pairs_in_both_directions(self):
constraints = [(('A', 'B'), self.constraint)]
arcs_result = all_arcs(constraints)
arcs_expected = set([('A', 'B'),
('B', 'A')])
self.assertEqual(arcs_result, arcs_expected)
def test_all_arcs(self):
# does not check that the constraint function is well created
variables = map(fst, list(all_arcs(self.constraints)))
arcs = [('A1', 'B2'),
('B2', 'A1'),
('A1', 'C2'),
('C2', 'A1'),
('D2', 'A1'),
('A1', 'D2'),
('D2', 'C1'),
('C1', 'D2'),
('E2', 'A1'),
('A1', 'E2')]
self.assertEqual(variables, arcs)
