def test_connect_row_of_sections(self):
"""
Test special case where connections have to branch
Row of Sections is connected, starting from the middle
RandomConnector can not connect this in one path, but has to branch
"""
section0 = new_section((0, 0), (10, 10), self.level, self.rng)
section1 = new_section((11, 0), (20, 10), self.level, self.rng)
section2 = new_section((21, 0), (30, 10), self.level, self.rng)
section3 = new_section((31, 0), (40, 10), self.level, self.rng)
section4 = new_section((41, 0), (50, 10), self.level, self.rng)
mark_neighbours(section0, section1)
mark_neighbours(section1, section2)
mark_neighbours(section2, section3)
mark_neighbours(section3, section4)
sections = [section0, section1, section2, section3, section4]
connected_sections = self.connector.connect_sections(sections, section2) # noqa
for section in connected_sections:
assert_that(list(section_connections(section)),
has_length(greater_than(0)))
assert_that(is_connected(section))
def test_connect_row_of_sections(self):
"""
Test special case where connections have to branch
Row of Sections is connected, starting from the middle
RandomConnector can not connect this in one path, but has to branch
"""
section0 = new_section((0, 0), (10, 10), self.level, self.rng)
section1 = new_section((11, 0), (20, 10), self.level, self.rng)
section2 = new_section((21, 0), (30, 10), self.level, self.rng)
section3 = new_section((31, 0), (40, 10), self.level, self.rng)
section4 = new_section((41, 0), (50, 10), self.level, self.rng)
mark_neighbours(section0, section1)
mark_neighbours(section1, section2)
mark_neighbours(section2, section3)
mark_neighbours(section3, section4)
sections = [section0, section1, section2, section3, section4]
connected_sections = self.connector.connect_sections(
sections, section2) # noqa
for section in connected_sections:
assert_that(list(section_connections(section)),
has_length(greater_than(0)))
assert_that(is_connected(section))
def test_connect_two_sections(self):
"""
Test that two adjacent sections can be connected
"""
section1 = new_section((0, 0), (10, 5), self.level, self.rng)
section2 = new_section((0, 6), (10, 10), self.level, self.rng)
mark_neighbours(section1, section2)
sections = [section1, section2]
connected_sections = self.connector.connect_sections(sections)
assert_that(connected_sections[1],
is_(equal_to(
list(section_connections(connected_sections[0]))[0].connection)))
def test_connect_two_sections(self):
"""
Test that two adjacent sections can be connected
"""
section1 = new_section((0, 0), (10, 5), self.level, self.rng)
section2 = new_section((0, 6), (10, 10), self.level, self.rng)
mark_neighbours(section1, section2)
sections = [section1, section2]
connected_sections = self.connector.connect_sections(sections)
assert_that(
connected_sections[1],
is_(
equal_to(
list(section_connections(
connected_sections[0]))[0].connection)))
def test_connecting_2x2_grid(self):
"""
Test that 2x2 grid is fully connected
"""
section00 = new_section((0, 0), (5, 5), self.level, self.rng)
section10 = new_section((6, 0), (10, 5), self.level, self.rng)
section01 = new_section((0, 6), (5, 10), self.level, self.rng)
section11 = new_section((6, 6), (10, 10), self.level, self.rng)
mark_neighbours(section00, section10)
mark_neighbours(section00, section01)
mark_neighbours(section10, section11)
mark_neighbours(section01, section11)
sections = [section00, section10, section01, section11]
connected_sections = self.connector.connect_sections(sections)
assert_that(connected_sections, has_length(4))
for section in connected_sections:
assert_that(list(section_connections(section)), has_length(greater_than(0)))
assert_that(is_connected(section))
def test_connecting_2x2_grid(self):
"""
Test that 2x2 grid is fully connected
"""
section00 = new_section((0, 0), (5, 5), self.level, self.rng)
section10 = new_section((6, 0), (10, 5), self.level, self.rng)
section01 = new_section((0, 6), (5, 10), self.level, self.rng)
section11 = new_section((6, 6), (10, 10), self.level, self.rng)
mark_neighbours(section00, section10)
mark_neighbours(section00, section01)
mark_neighbours(section10, section11)
mark_neighbours(section01, section11)
sections = [section00, section10, section01, section11]
connected_sections = self.connector.connect_sections(sections)
assert_that(connected_sections, has_length(4))
for section in connected_sections:
assert_that(list(section_connections(section)),
has_length(greater_than(0)))
assert_that(is_connected(section))