def test_two_nine_tile_rooms(self):
"""Corridor is chosen randomly out of possible options."""
room_a, room_b = Room(width=3, height=3), Room(width=3, height=3)
room_a.x, room_a.y, room_b.x, room_b.y = 0, 0, 4, 1
self.assertIn(
_get_corridor_coordinates(room_a, room_b),
set([((3, 1), ((2, 1), (0, 0)), (1, 0)),
((3, 2), ((2, 2), (0, 1)), (1, 0))]))
def test_two_one_tile_rooms_from_north(self):
room_a, room_b = Room(width=1, height=1), Room(width=1, height=1)
room_a.x, room_a.y, room_b.x, room_b.y = 0, 2, 0, 0
self.assertEqual(_get_corridor_coordinates(room_a, room_b),
((0, -1), ((0, 0), (0, 0)), (0, -1)))
def test_rooms_are_too_far_away(self):
room_a, room_b = Room(width=1, height=1), Room(width=1, height=1)
room_a.x, room_a.y, room_b.x, room_b.y = 0, 0, 9, 9
self.assertEqual(_get_corridor_coordinates(room_a, room_b),
(None, (None, None), None))
