class TestBasic(TestCase):
def setUp(self):
self.links = [
(1, 10, 100),
(1, 20, 200),
(2, 10, 100),
(2, 10, 200),
(2, 20, 100),
(2, 20, 200),
]
self.constraint_links = ConstraintLinksBase(self.links)
def test_distinct_constraint_ids(self):
assert set(self.constraint_links.distinct_constraint_ids()) == set(
[100, 200])
def test_add_links(self):
new_links = [(2, 20, 200), (3, 30, 300)]
self.constraint_links.add_links(new_links)
constr_links = self.constraint_links.get_links()
assert len(self.links) + 1 == len(constr_links)
assert set(constr_links) == set([
(1, 10, 100),
(1, 20, 200),
(2, 10, 100),
(2, 10, 200),
(2, 20, 100),
(2, 20, 200),
(3, 30, 300),
])
def test_remove_by_line(self):
self.constraint_links.remove_links_by_line(1)
assert set(self.constraint_links.get_links()) == set([
(2, 10, 100),
(2, 10, 200),
(2, 20, 100),
(2, 20, 200),
])
def test_remove_by_group(self):
self.constraint_links.remove_links_by_group(2, 20)
assert set(self.constraint_links.get_links()) == set([
(1, 10, 100),
(1, 20, 200),
(2, 10, 100),
(2, 10, 200),
])
def test_remove_by_constraint(self):
self.constraint_links.remove_links_by_constraint(100)
assert set(self.constraint_links.get_links()) == set([
(1, 20, 200),
(2, 10, 200),
(2, 20, 200),
])
class TestBasic(TestCase):
def setUp(self):
self.links = [
(1, 10, 100),
(1, 20, 200),
(2, 10, 100),
(2, 10, 200),
(2, 20, 100),
(2, 20, 200),
]
self.constraint_links = ConstraintLinksBase(self.links)
def test_distinct_constraint_ids(self):
assert set(self.constraint_links.distinct_constraint_ids()) == set([100, 200])
def test_add_links(self):
new_links = [(2, 20, 200), (3, 30, 300)]
self.constraint_links.add_links(new_links)
constr_links = self.constraint_links.get_links()
assert len(self.links) + 1 == len(constr_links)
assert set(constr_links) == set(
[
(1, 10, 100),
(1, 20, 200),
(2, 10, 100),
(2, 10, 200),
(2, 20, 100),
(2, 20, 200),
(3, 30, 300),
]
)
def test_remove_by_line(self):
self.constraint_links.remove_links_by_line(1)
assert set(self.constraint_links.get_links()) == set(
[
(2, 10, 100),
(2, 10, 200),
(2, 20, 100),
(2, 20, 200),
]
)
def test_remove_by_group(self):
self.constraint_links.remove_links_by_group(2, 20)
assert set(self.constraint_links.get_links()) == set(
[
(1, 10, 100),
(1, 20, 200),
(2, 10, 100),
(2, 10, 200),
]
)
def test_remove_by_constraint(self):
self.constraint_links.remove_links_by_constraint(100)
assert set(self.constraint_links.get_links()) == set(
[
(1, 20, 200),
(2, 10, 200),
(2, 20, 200),
]
)