fix job exporter test issue

microsoft · Aug 6, 2020 · 45253ed · 45253ed
1 parent f1b98e1
commit 45253ed
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/job-exporter/test/test_collector.py b/src/job-exporter/test/test_collector.py
@@ -48,7 +48,8 @@ def test_parse_from_labels(self):
                 "0",
                 "0,1,",
                 "application_1522829300813_1943",
-                12345)
+                12345,
+                "default")
 
         gpu_ids, labels = ContainerCollector.parse_from_labels(inspect_result, None)
         self.assertEqual(["0", "1"], gpu_ids)
@@ -58,7 +59,8 @@ def test_parse_from_labels(self):
                 "job_name": "trialslot_nnimain_d65bc5ac",
                 "role_name": "tuner",
                 "task_index": "0",
-                "job_instance_id": "application_1522829300813_1943"}
+                "job_instance_id": "application_1522829300813_1943",
+                "virtual_cluster":"default"}
 
         self.assertEqual(target_labels, labels)