binbot1 = Bin("robot_4", 3, 0.5, -8.75, -39, pi / 2) binbot2 = Bin("robot_5", 3, 0.5, -1.75, -39, pi / 2) binbot3 = Bin("robot_6", 3, 0.5, 1.75, -39, pi / 2) binbot4 = Bin("robot_7", 3, 0.5, 5.25, -39, pi / 2) picker1 = PickerRobot("robot_8", 3, 0.5, -8.75, -44, 0) picker2 = PickerRobot("robot_9", 3, 0.5, 1, -44, 0) picker3 = PickerRobot("robot_10", 3, 0.5, 7, -44, 0) carrier1 = CarrierRobot("robot_11", 3, 0.5, 35.5, -25, 0) carrier2 = CarrierRobot("robot_12", 2, 0.5, 41.5, -35, 0) # Add all robots to robot_storage. robot_storage.addRobot(animal, "robot_0") robot_storage.addRobot(person1, "robot_1") robot_storage.addRobot(person2, "robot_2") robot_storage.addRobot(tractor, "robot_3") robot_storage.addRobot(binbot1, "robot_4") robot_storage.addRobot(binbot2, "robot_5") robot_storage.addRobot(binbot3, "robot_6") robot_storage.addRobot(binbot4, "robot_7") robot_storage.addRobot(picker1, "robot_8") robot_storage.addRobot(picker2, "robot_9") robot_storage.addRobot(picker3, "robot_10") robot_storage.addRobot(carrier1, "robot_11") robot_storage.addRobot(carrier2, "robot_12") rate = rospy.Rate(10)
binbot1 = Bin('robot_4', 3, 0.5, -8.75, -39, pi/2) binbot2 = Bin('robot_5', 3, 0.5, -1.75, -39, pi/2) binbot3 = Bin('robot_6', 3, 0.5, 1.75, -39, pi/2) binbot4 = Bin('robot_7', 3, 0.5, 5.25, -39, pi/2) picker1 = PickerRobot('robot_8', 3, 0.5, -8.75, -44, 0) picker2 = PickerRobot('robot_9', 3, 0.5, 1, -44, 0) picker3 = PickerRobot('robot_10', 3, 0.5, 7, -44, 0) carrier1 = CarrierRobot('robot_11', 3, 0.5, 35.5, -25, 0) carrier2 = CarrierRobot('robot_12', 2, 0.5, 41.5, -35, 0) # Add all robots to robot_storage. robot_storage.addRobot(animal, "robot_0") robot_storage.addRobot(person1, "robot_1") robot_storage.addRobot(person2, "robot_2") robot_storage.addRobot(tractor, "robot_3") robot_storage.addRobot(binbot1, "robot_4") robot_storage.addRobot(binbot2, "robot_5") robot_storage.addRobot(binbot3, "robot_6") robot_storage.addRobot(binbot4, "robot_7") robot_storage.addRobot(picker1, "robot_8") robot_storage.addRobot(picker2, "robot_9") robot_storage.addRobot(picker3, "robot_10") robot_storage.addRobot(carrier1, "robot_11") robot_storage.addRobot(carrier2, "robot_12") rate = rospy.Rate(10)
def test_get_robot_with_id(self): """Test get robot with id method returns robot""" robot_storage.addRobot(self.robot_obj, self.robot_obj.robot_id) result = robot_storage.getRobotWithId(self.robot_obj.robot_id) self.assertEqual(result, self.robot_obj)
def test_add_robot(self): """test add robot method""" robot_storage.addRobot(self.robot_obj, self.robot_obj.robot_id) test_list = robot_storage.get_robot_list() self.assertIn(self.robot_obj.robot_id, test_list)