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)