def gen_policy(self):
self.policy = make_job_resource_policy(
self.sku,
1, # 1 GPU per worker
self.config,
self.quota,
self.metadata)
def test_cpu_job(self):
policy = make_job_resource_policy("Standard_ND24rs", 0, self.config,
self.quota, self.metadata)
self.assertIsNotNone(policy)
self.assertEqual("1000m", policy.default_cpu_request)
self.assertEqual("24000m", policy.default_cpu_limit)
self.assertEqual("0Mi", policy.default_memory_request)
self.assertEqual("458752Mi", policy.default_memory_limit)
def test_gpu_job(self):
policy = make_job_resource_policy("Standard_ND24rs", 3, self.config,
self.quota, self.metadata)
self.assertIsNotNone(policy)
self.assertEqual("16000m", policy.default_cpu_request)
self.assertEqual("18000m", policy.default_cpu_limit)
self.assertEqual("309657Mi", policy.default_memory_request)
self.assertEqual("344064Mi", policy.default_memory_limit)
def test_override(self):
self.config.update({
"default_cpurequest": "12000m",
"default_cpulimit": "14000m",
"default_memoryrequest": "102400Mi",
"default_memorylimit": "409600Mi",
})
policy = make_job_resource_policy("Standard_ND24rs", 0, self.config,
self.quota, self.metadata)
self.assertIsNotNone(policy)
self.assertEqual("12000m", policy.default_cpu_request)
self.assertEqual("14000m", policy.default_cpu_limit)
self.assertEqual("102400Mi", policy.default_memory_request)
self.assertEqual("409600Mi", policy.default_memory_limit)
def gen_policy(self):
self.policy = make_job_resource_policy(self.sku, self.gpu_limit,
self.config, self.quota,
self.metadata)