def test_infer_service_name(self): self.assertIsNone(ContainerCollector.infer_service_name( "k8s_POD_alertmanager-7884c59f78-66r86_default_0a32e30a-f6ae-11e8")) self.assertEqual( "alertmanager", ContainerCollector.infer_service_name( "k8s_alertmanager_alertmanager-7884c59f78-66r86_default_0a32e30a-f6ae-11e8-a62d-000d3ab25bb6_2")) self.assertIsNone(ContainerCollector.infer_service_name( "k8s_kube-scheduler_kube-scheduler-10.151.40.4_kube-system_f1164d931979939cf0601155df9c748a_6"))
def test_parse_from_labels(self): inspect_result = docker_inspect.InspectResult( "openmindstudio", "trialslot_nnimain_d65bc5ac", "tuner", "0", "this_is_pod_name_val", "0,1,", 12345, "*****@*****.**", "platform", False, False, ) gpu_ids, labels = ContainerCollector.parse_from_labels( inspect_result, None) self.assertEqual(["0", "1"], gpu_ids) target_labels = { "username": "******", "job_name": "trialslot_nnimain_d65bc5ac", "role_name": "tuner", "task_index": "0", "pod_name": "this_is_pod_name_val", "user_email": "*****@*****.**", "vc_name": "platform", "preemptible": "false", } self.assertEqual(target_labels, labels)
def test_parse_from_labels(self): labels = { "container_label_PAI_USER_NAME": "openmindstudio", "container_label_GPU_ID": "0,1,", "container_label_PAI_JOB_NAME": "trialslot_nnimain_d65bc5ac", "container_label_PAI_CURRENT_TASK_ROLE_NAME": "tuner" } gpuIds, otherLabels = ContainerCollector.parse_from_labels(labels) self.assertEqual( ["0", "1"], gpuIds, ) copied = copy.deepcopy(labels) copied.pop("container_label_GPU_ID") self.assertEqual(copied, otherLabels)
def test_parse_from_labels(self): inspect_result = docker_inspect.InspectResult( "openmindstudio", "trialslot_nnimain_d65bc5ac", "tuner", "0", "0,1,", 12345) gpu_ids, labels = ContainerCollector.parse_from_labels( inspect_result, None) self.assertEqual(["0", "1"], gpu_ids) target_labels = { "username": "******", "job_name": "trialslot_nnimain_d65bc5ac", "role_name": "tuner", "task_index": "0" } self.assertEqual(target_labels, labels)
def test_parse_from_labels(self): inspect_result = docker_inspect.InspectResult( "openmindstudio", "trialslot_nnimain_d65bc5ac", "tuner", "0", "0,1,", "application_1522829300813_1943", "default", 12345) gpu_ids, labels = ContainerCollector.parse_from_labels( inspect_result, None) self.assertEqual(["0", "1"], gpu_ids) target_labels = { "username": "******", "job_name": "trialslot_nnimain_d65bc5ac", "role_name": "tuner", "task_index": "0", "job_instance_id": "application_1522829300813_1943", "virtual_cluster": "default" } self.assertEqual(target_labels, labels)