fix test

m3dev · Sep 28, 2024 · c402f75 · c402f75
1 parent 6f8772d
commit c402f75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_task_on_kart.py b/test/test_task_on_kart.py
@@ -267,7 +267,7 @@ def test_load_with_single_dict_target(self):
 
         data = task.load()
         target.load.assert_called_once()
-        self.assertEqual(data, 1)
+        self.assertEqual(data, {'target_key': 1})
 
     def test_load_with_keyword(self):
         task = _DummyTask()