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)