def test_maze_walls(self): maze = Maze(width=1, height=1) maze._grid[0][0] = Compass.NORTH | Compass.WEST self.assertEqual(maze.walls(0, 0), {Compass.NORTH, Compass.WEST})