def test_get_status_READY(self): maze = Maze() length = randint(1, 10) maze.set_board([[Cell(i, j, True) for i in range(length)] for j in range(length)]) maze.set_track({maze.get_cell(0, 0): []}) self.assertEqual(MazeStatus.READY, maze.get_status())
def test_get_status_GENERATING(self): maze = Maze() length = randint(1, 10) maze.set_board([[Cell(i, j, False) for i in range(length)] for j in range(length)]) maze.set_track({maze.get_cell(0, 0): []}) maze.set_cell_value(0, 0, True) self.assertEqual(MazeStatus.GENERATING, maze.get_status())