update unittesting

ArkMowers · Feb 16, 2025 · ba28dd2 · ba28dd2
1 parent 96b96db
commit ba28dd2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/arknights_mower/tests/scheduler_task_tests.py b/arknights_mower/tests/scheduler_task_tests.py
@@ -377,7 +377,7 @@ def test_reorder_1(self):
         op_data.operators["凯尔希"].current_room = "dormitory_2"
         op_data.operators["凯尔希"].current_index = 2
         op_data.dorm[2].name = "夕"
-        plan = try_reorder(op_data)
+        plan = try_reorder(op_data, {})
         self.assertEqual(plan["dormitory_1"][2], "夕")
         tasks = [SchedulerTask(task_plan=plan, task_type=TaskTypes.SHIFT_OFF)]
         check_dorm_ordering(tasks, op_data)
@@ -393,7 +393,7 @@ def test_reorder_2(self):
         op_data.dorm[4].name = "森蚺"
 
         # op_data.config.ope_resting_priority=["森蚺","夕"]
-        plan = try_reorder(op_data)
+        plan = try_reorder(op_data, {})
         self.assertEqual(len(plan), 3)
         self.assertEqual(plan["dormitory_1"][2], "夕")
         self.assertEqual(plan["dormitory_1"][4], "凯尔希")
@@ -411,8 +411,8 @@ def test_reorder_3(self):
         op_data.operators["见行者"].current_room = "dormitory_2"
         op_data.operators["见行者"].current_index = 2
         op_data.dorm[4].name = "见行者"
-        try_reorder(op_data)
-        plan = try_reorder(op_data)
+        try_reorder(op_data, {})
+        plan = try_reorder(op_data, {})
         self.assertEqual(plan["dormitory_1"][2], "夕")
         self.assertEqual(plan["dormitory_1"][3], "见行者")
         tasks = [SchedulerTask(task_plan=plan, task_type=TaskTypes.SHIFT_OFF)]