def test_adapt_dlts_jobs(self):
sample_path = "data/dlts_docker_inspect.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect, GpuVendor.NVIDIA)
target_inspect_info = InspectResult(
"dixu", "0c435eee-d31f-43d5-a1b3-442845fa1d0c", None, None,
"GPU-7c583998-b3ff-a885-8979-2d32d334cde4", None, None, 3533)
self.assertEqual(target_inspect_info, inspect_info)
def test_parse_docker_inspect(self):
sample_path = "data/docker_inspect_kube_launcher_task.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect)
target_inspect_info = InspectResult(
"core", "core~tensorflowcifar10", "worker", "0",
"GPU-dc0671b0-61a4-443e-f456-f8fa6359b788", 23774)
self.assertEqual(target_inspect_info, inspect_info)
def test_parse_docker_inspect_amd(self):
sample_path = "data/docker_inspect_amd.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect, GpuVendor.AMD)
target_inspect_info = InspectResult(
"dgxadmin", "dgxadmin~rocm_pytorch_mnist_a", "worker", "0", "0,1",
"0_a12d82af-5ea6-11ea-8a2a-90b11c27f535", "default", 29647)
self.assertEqual(target_inspect_info, inspect_info)
def test_parse_docker_inspect(self):
sample_path = "data/docker_inspect_sample.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect)
target_inspect_info = InspectResult("openmindstudio",
"trialslot_nnimain_d65bc5ac",
"tuner", "0", "0,1,", 95539)
self.assertEqual(target_inspect_info, inspect_info)
def test_parse_docker_inspect_kube(self):
sample_path = "data/docker_inspect_kube_launcher_task.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect, GpuVendor.NVIDIA)
target_inspect_info = InspectResult(
"core", "core~tensorflowcifar10", "worker", "0",
"GPU-dc0671b0-61a4-443e-f456-f8fa6359b788",
"0_69c05215-46fa-11e9-8937-000d3ab38724", "default", 23774)
self.assertEqual(target_inspect_info, inspect_info)
def test_parse_docker_inspect_BUGFIX(self):
sample_path = "data/inspect_result_bug_fix.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect)
target_inspect_info = InspectResult(
"sokoya",
"sokoya~train-exp_offrl_sc_discard_0231-10th-beta07-lrfixed_13e9bf5_gCYv",
"train", "0", "3,2,1,0", "application_1553664769226_0080", 30332)
self.assertEqual(target_inspect_info, inspect_info)
def test_parse_docker_inspect(self):
sample_path = "data/docker_inspect_sample.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect, GpuVendor.NVIDIA)
target_inspect_info = InspectResult("openmindstudio",
"trialslot_nnimain_d65bc5ac",
"tuner", "0", "0,1,",
"application_1522829300813_1943",
"default", 95539)
self.assertEqual(target_inspect_info, inspect_info)
def test_parse_docker_inspect(self):
sample_path = "data/docker_inspect_kube_launcher_task.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect)
target_inspect_info = {
"labels": {
"container_label_PAI_USER_NAME": "core",
"container_label_PAI_JOB_NAME": "core~tensorflowcifar10",
"container_label_PAI_CURRENT_TASK_ROLE_NAME": "worker"
},
"env": {
"container_env_PAI_TASK_INDEX": "0"
},
"pid": 23774
}
self.assertEqual(target_inspect_info, inspect_info)
def test_parse_docker_inspect(self):
sample_path = "data/docker_inspect_sample.json"
with open(sample_path, "r") as f:
docker_inspect = f.read()
inspect_info = parse_docker_inspect(docker_inspect)
target_inspect_info = {
"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"
},
"env": {
"container_env_PAI_TASK_INDEX": "0"
},
"pid": 95539
}
self.assertEqual(target_inspect_info, inspect_info)
