def test_set_state(self):
maze_cells = np.array([
[Maze.EMPTY, Maze.S, Maze.EMPTY, Maze.E, Maze.EMPTY],
[Maze.EMPTY, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY],
[Maze.N, Maze.N, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY],
[Maze.EMPTY, Maze.EMPTY, Maze.S, Maze.EMPTY, Maze.EMPTY],
[Maze.E, Maze.E, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY]
])
maze = Maze(maze_cells)
robots = {
"A": (0, 2),
"B": (0, 2),
"C": (4, 2)
}
goals = queue.Queue()
goals.put(Goal("A", (0, 0)))
goals.put(Goal("B", (0, 0)))
goals.put(Goal("C", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
rr.move_robot("A", RobotReboot.MOVE_NORTH)
rr.move_robot("B", RobotReboot.MOVE_EAST)
other_goals = queue.Queue()
other_goals.put(Goal("A", (0, 1)))
another_rr = RobotReboot(Maze(np.array([[0, 0, 0, 0, 0]])), other_goals)
another_rr.set_game(["A", "B", "C"], rr.state(), rr.current_game.movements)
self.assertEqual(rr.robots, another_rr.robots)
self.assertEqual(rr.state().all(), another_rr.state().all())
self.assertListEqual(rr.current_game.movements, another_rr.current_game.movements)
self.assertIsNotNone(rr.maze)
def test_reset(self):
maze_cells = np.array([
[Maze.EMPTY, Maze.S, Maze.EMPTY, Maze.E, Maze.EMPTY],
[Maze.EMPTY, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY],
[Maze.N, Maze.N, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY],
[Maze.EMPTY, Maze.EMPTY, Maze.S, Maze.EMPTY, Maze.EMPTY],
[Maze.E, Maze.E, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY]
])
maze = Maze(maze_cells)
goals = queue.Queue()
goals.put(Goal("A", (0, 1)))
goals.put(Goal("B", (3, 4)))
goals.put(Goal("C", (0, 5)))
rr = RobotReboot(maze, goals)
init_robots = rr.robots_initial.copy()
rr.move_robot("A", RobotReboot.MOVE_SOUTH)
rr.move_robot("B", RobotReboot.MOVE_EAST)
rr.move_robot("C", RobotReboot.MOVE_NORTH)
self.assertNotEqual(rr.robots, init_robots)
rr.reset()
self.assertEqual(rr.robots, init_robots)
def test_move_robot_west_when_another_robot_on_the_way(self):
maze_cells = np.array([[0, 0, 0, 0, 0]])
maze = Maze(maze_cells)
robots = {
"A": (0, 4),
"B": (0, 2)
}
goals = queue.Queue()
goals.put(Goal("A", (0, 0)))
goals.put(Goal("B", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
rr.move_robot("A", RobotReboot.MOVE_WEST)
self.assertEqual((0, 3), rr.robots["A"])
def test_move_robot_north_when_another_robot_on_the_way(self):
maze_cells = np.array([
[0],
[0],
[0],
[0],
[0]
])
maze = Maze(maze_cells)
robots = {
"A": (4, 0),
"B": (2, 0)
}
goals = queue.Queue()
goals.put(Goal("A", (0, 0)))
goals.put(Goal("B", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
rr.move_robot("A", RobotReboot.MOVE_NORTH)
self.assertEqual((3, 0), rr.robots["A"])
def test_state(self):
maze_cells = np.array([
[Maze.EMPTY, Maze.S, Maze.EMPTY, Maze.E, Maze.EMPTY],
[Maze.EMPTY, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY],
[Maze.N, Maze.N, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY],
[Maze.EMPTY, Maze.EMPTY, Maze.S, Maze.EMPTY, Maze.EMPTY],
[Maze.E, Maze.E, Maze.EMPTY, Maze.EMPTY, Maze.EMPTY]
])
maze = Maze(maze_cells)
robots = {
"A": (0, 2),
"B": (0, 2),
"C": (4, 2)
}
goals = queue.Queue()
# Order here matters, first robot is B, next robot is A, last robot is C
goals.put(Goal("B", (3, 4)))
goals.put(Goal("A", (0, 0)))
goals.put(Goal("C", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
obs = rr.state()
rows, cols, layers = obs.shape
self.assertEqual(obs.shape, (5, 5, 4))
np.testing.assert_equal(obs[:, :, 0], maze_cells)
# Checking robots on each layer
self.assertEqual(obs[0, 2, 1], RobotReboot.ROBOT)
self.assertEqual(obs[0, 2, 2], RobotReboot.ROBOT)
self.assertEqual(obs[4, 2, 3], RobotReboot.ROBOT)
# Checking goal on the target robot
self.assertEqual(obs[3, 4, 1], RobotReboot.GOAL)
for i in range(rows):
for j in range(cols):
for layer in range(1, layers):
if not rr.is_a_robot_on((i, j)) and rr.goal.cell != (i, j) and layer != 2:
self.assertEqual(obs[i, j, layer], 0)
def test_is_a_robot_on_position_on_maze(self):
maze_cells = np.array([[0, 0, 0, 0, 0]])
maze = Maze(maze_cells)
robots = {
"A": (0, 2),
}
goals = queue.Queue()
goals.put(Goal("A", (0, 1)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
self.assertTrue(rr.is_a_robot_on((0, 2)))
self.assertFalse(rr.is_a_robot_on((0, 0)))
def test_move_robot_west_when_robot_at_border(self):
maze_cells = np.array([[0, 0, 0, 0, 0]])
maze = Maze(maze_cells)
robots = {
"A": (0, 0)
}
goals = queue.Queue()
goals.put(Goal("A", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
rr.move_robot("A", RobotReboot.MOVE_WEST)
self.assertEqual((0, 0), rr.robots["A"])
def test_move_robot_east_with_wall_at_west(self):
maze_cells = np.array([[0, 0, Maze.W, 0, 0]])
maze = Maze(maze_cells)
robots = {
"A": (0, 0)
}
goals = queue.Queue()
goals.put(Goal("A", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
rr.move_robot("A", RobotReboot.MOVE_EAST)
self.assertEqual((0, 1), rr.robots["A"])
def test_move_robot_north_with_wall_at_north(self):
maze_cells = np.array([
[0],
[0],
[Maze.N],
[0],
[0]
])
maze = Maze(maze_cells)
robots = {
"A": (4, 0)
}
goals = queue.Queue()
goals.put(Goal("A", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
rr.move_robot("A", RobotReboot.MOVE_NORTH)
self.assertEqual((2, 0), rr.robots["A"])
def test_state_win_game(self):
maze_cells = np.array([
[Maze.EMPTY, Maze.S, Maze.EMPTY, Maze.S, Maze.EMPTY]
])
maze = Maze(maze_cells)
robots = {
"A": (0, 4),
}
goals = queue.Queue()
# Order here matters, first robot is B, next robot is A, last robot is C
goals.put(Goal("A", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
rr.move_robot("A", RobotReboot.MOVE_WEST)
obs = rr.state()
self.assertEqual(obs[0, 0, 1], RobotReboot.GOAL_ROBOT)
def test_move_robot_south_when_wall_at_west_and_east(self):
maze_cells = np.array([
[0],
[Maze.E],
[Maze.W],
[0],
[0]
])
maze = Maze(maze_cells)
robots = {
"A": (0, 0)
}
goals = queue.Queue()
goals.put(Goal("A", (0, 0)))
rr = RobotReboot(maze, goals)
rr.set_robots(robots)
rr.move_robot("A", RobotReboot.MOVE_SOUTH)
self.assertEqual((4, 0), rr.robots["A"])
