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)
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)
