def test_empty_observation(self): robot = GripperRobot(10, Heading.NORTH, (5, 5)) game = self.default_game10x10() game.grid = np.zeros((10, 10)) observation = robot.get_observation(game) self.assertEqual(17, len(observation)) for i in range(len(observation)): self.assertEqual(0, observation[i])
def test_robot_observation_1(self): robot = GripperRobot(10, Heading.NORTH, (5, 5)) game = self.default_game10x10() game.grid = np.zeros((10, 10)) game.robots.append(GripperRobot(11, location=(6, 5))) observation = robot.get_observation(game) for i in range(16): self.assertEqual(0, observation[i]) self.assertEqual(1, observation[16])
def test_robot_observation_5(self): robot = GripperRobot(10, Heading.NORTH, (5, 5)) game = self.default_game10x10() game.grid = np.zeros((10, 10)) game.robots.append(GripperRobot(11, location=(4, 4))) observation = robot.get_observation(game) self.assertEqual(1, observation[7]) for i in [0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 16]: self.assertEqual(0, observation[i])
def test_border_observation_3(self): robot = GripperRobot(10, Heading.EAST, (9, 0)) game = self.default_game10x10() game.grid = np.zeros((10, 10)) observation = robot.get_observation(game) for i in [0, 1, 3, 4, 6, 7, 9, 10, 12, 13, 14, 15, 16]: self.assertEqual(0, observation[i]) for i in [2, 5, 8, 11]: self.assertEqual(1, observation[i])
def test_tile_everywhere(self): robot = GripperRobot(10, Heading.NORTH, (5, 5)) game = self.default_game10x10() game.grid = np.zeros((10, 10)) game.grid[4][5] = 1 game.grid[6][5] = 1 game.grid[4][4] = 1 game.grid[5][4] = 1 game.grid[6][4] = 1 observation = robot.get_observation(game) for i in [3, 6, 9, 12, 15]: self.assertEqual(1, observation[i]) for i in [0, 1, 2, 4, 5, 7, 8, 10, 11, 13, 14, 16]: self.assertEqual(0, observation[i])
def test_robot_everywhere(self): robot = GripperRobot(10, Heading.SOUTH, (5, 5)) game = self.default_game10x10() game.grid = np.zeros((10, 10)) game.robots.append(GripperRobot(11, location=(4, 5))) game.robots.append(GripperRobot(12, location=(6, 5))) game.robots.append(GripperRobot(13, location=(4, 6))) game.robots.append(GripperRobot(14, location=(5, 6))) game.robots.append(GripperRobot(15, location=(6, 6))) observation = robot.get_observation(game) for i in [4, 7, 10, 13, 16]: self.assertEqual(1, observation[i]) for i in [0, 1, 2, 3, 5, 6, 8, 9, 11, 12, 14, 15]: self.assertEqual(0, observation[i])