def test_shift_horizontal(self):
r = Room(self.cells)
r.shift_horizontal(100)
for cell in r.cells:
assert cell.coordinates[1] >= 100, cell.coordinates
for i in range(4):
for j in range(5):
assert (i, j + 100) in r.cell_coordinates
def test_shift_horizontal_one_col(self):
r = Room(self.cells)
r.shift_horizontal(1)
for i in range(4):
for j in range(5):
assert (i, j + 1) in r.cell_coordinates