def test_format_place_task(self):
print ("Testing method: " + str(self._testMethodName))
tm = TaskManager()
result = ActionList()
t = Task(t_type=1, source=2, destination=4, container=2, t_object=4)
result.action_list = tm.format_place_task(t, result.action_list)
action_key = TaskProtocol.look_up_value(TaskProtocol.task_action_dict, "PLACE")
self.assertEqual(len(result.action_list), 1)
for item in result.action_list:
self.assertEqual(item.action, action_key)
self.assertEqual(item.action_str, "PLACE")
self.assertEqual(item.type, t.type)
self.assertEqual(item.type_str, t.type_str)
self.assertEqual(item.source, t.source)
self.assertEqual(item.source_str, t.source_str)
self.assertEqual(item.destination, t.destination)
self.assertEqual(item.destination_str, t.destination_str)
self.assertEqual(item.object, t.object)
self.assertEqual(item.object_str, t.object_str)
self.assertEqual(item.container, t.container)
self.assertEqual(item.container_str, t.container_str)
def test_format_set_of_tasks(self):
print ("Testing method: " + str(self._testMethodName))
tm = TaskManager()
tl = TaskList()
t1 = Task(t_type=1, source=2, destination=4, t_object=4)
tl.add_task(t1)
t2 = Task(t_type=1, source=3, destination=3, t_object=2)
tl.add_task(t2)
t3 = Task(t_type=1, source=2, destination=4, t_object=3)
tl.add_task(t3)
result = ActionList()
result.action_list = tm.format_set_of_tasks(tl, result.action_list)
def test_format_pick_from_robot(self):
print ("Testing method: " + str(self._testMethodName))
tm = TaskManager()
result = ActionList()
t = Task(t_type=1, source=2, destination=4, container=2, t_object=4)
result.action_list = tm.format_pick_from_robot(t, result.action_list)
action_key = TaskProtocol.look_up_value(TaskProtocol.task_action_dict, "PICK_FROM_ROBOT")
self.assertEqual(len(result.action_list), 1)
for item in result.action_list:
self.assertEqual(item.action, action_key)
self.assertEqual(item.action_str, "PICK_FROM_ROBOT")
self.assertEqual(item.object, t.object)
self.assertEqual(item.object_str, t.object_str)
def test_format_drive(self):
print ("Testing method: " + str(self._testMethodName))
tm = TaskManager()
result = ActionList()
dest_key = TaskProtocol.look_up_value(TaskProtocol.location_dict, "Workstation 1")
action_key = TaskProtocol.look_up_value(TaskProtocol.task_action_dict, "DRIVE")
result.action_list = tm.format_drive(dest_key, result.action_list)
self.assertEqual(len(result.action_list), 1)
for item in result.action_list:
self.assertEqual(item.action, action_key)
self.assertEqual(item.action_str, "DRIVE")
self.assertEqual(item.destination, dest_key)
self.assertEqual(item.destination_str, "Workstation 1")
def test_optimize_list(self):
print ("Testing method: " + str(self._testMethodName))
tm = TaskManager()
tl = TaskList()
t1 = Task(t_type=2, source=2, destination=1, t_object=4)
tl.add_task(t1)
t2 = Task(t_type=2, source=3, destination=3, t_object=2)
tl.add_task(t2)
t3 = Task(t_type=2, source=2, destination=4, t_object=3)
tl.add_task(t3)
t4 = Task(t_type=2, source=5, destination=4, t_object=3)
tl.add_task(t4)
result = ActionList()
result.action_list = tm.optimize_list(tl, result.action_list)
print(result)
def test_format_pick (self):
print ("Testing method: " + str(self._testMethodName))
tm = TaskManager()
tl = TaskList()
t1 = Task(t_type=1, source=2, destination=4, container=2, t_object=4)
tl.add_task(t1)
t2 = Task(t_type=1, source=2, destination=3, container=5, t_object=2)
tl.add_task(t2)
t3 = Task(t_type=1, source=3, destination=4, container=2, t_object=3)
tl.add_task(t3)
result = ActionList()
result.action_list = tm.format_pick(tl, result.action_list)
# If reached here, format_pick_task passed the test
# So only assert len
self.assertEquals(len(result.action_list), 3)