def test_move_off_screen_test_2(self): robot = GripperRobot(10, Heading.EAST, (1, 0)) game = TilingPatternGame((1, 1), 2, RandomRobotReset(GripperRobot, 0), RandomWorldReset()) robot.step([True, 0, False, False], game) self.assertEqual((1, 0), robot.location)
def test_pickup_none_action(self): robot = GripperRobot(10, Heading.WEST, (1, 1)) game = self.default_game() game.grid[1][1] = False robot.step([False, 0, True, False], game) self.assertFalse(robot.hold_object) self.assertFalse(game.grid[1][1])
def test_drop_occupied_action(self): robot = GripperRobot(10, Heading.WEST) robot.hold_object = True game = self.default_game() game.grid[0][0] = True robot.step([False, 0, False, True], game) self.assertTrue(robot.hold_object) self.assertTrue(game.grid[0][0])
def test_drop_action(self): robot = GripperRobot(10, Heading.WEST, (1, 1)) robot.hold_object = True game = self.default_game() game.grid[1][1] = False robot.step([False, 0, False, True], game) self.assertFalse(robot.hold_object) self.assertTrue(game.grid[1][1])
def test_pickup_action(self): robot = GripperRobot(10, Heading.WEST) game = TilingPatternGame((1, 1), 1, RandomRobotReset(GripperRobot, 0), RandomWorldReset()) game.reset() self.assertTrue(game.grid[0][0]) robot.step([False, 0, True, False], game) self.assertTrue(robot.hold_object) self.assertFalse(game.grid[0][0])
def test_rotate_robot_8_action(self): robot = GripperRobot(10, Heading.WEST) game = self.default_game() robot.step([False, -1, False, False], game) self.assertEqual(Heading.SOUTH, robot.heading)
def test_rotate_robot_2_action(self): robot = GripperRobot(10, Heading.NORTH) game = self.default_game() robot.step([False, 0, False, False], game) self.assertEqual(Heading.NORTH, robot.heading)
def test_move_off_screen_test_3(self): robot = GripperRobot(10, Heading.SOUTH, (1, 1)) game = self.default_game() robot.step([True, 0, False, False], game) self.assertEqual((1, 1), robot.location)
def test_move_forward_test_3(self): robot = GripperRobot(10, Heading.NORTH, (0, 1)) game = self.default_game() robot.step([True, 0, False, False], game) self.assertEqual((0, 0), robot.location)