def testCustomMachineTypeFlagOverride(self):
flags = mock_flags.MockFlags()
flags['machine_type'].parse('{cpus: 1, memory: 7.5GiB}')
result = gce_virtual_machine.GceVmSpec(
_COMPONENT, flag_values=flags, machine_type='n1-standard-8')
self.assertEqual(result.machine_type, None)
self.assertEqual(result.cpus, 1)
self.assertEqual(result.memory, 7680)
def testMissingGpuCount(self):
flags = mock_flags.MockFlags()
with self.assertRaises(errors.Config.MissingOption) as cm:
virtual_machine.BaseVmSpec(_COMPONENT,
flag_values=flags,
gpu_type='k80')
self.assertEqual(str(cm.exception),
('gpu_count must be specified if gpu_type is set'))
def testStringMachineTypeFlagOverride(self):
flags = mock_flags.MockFlags()
flags['machine_type'].parse('n1-standard-8')
result = gce_virtual_machine.GceVmSpec(
_COMPONENT, flag_values=flags,
machine_type={'cpus': 1, 'memory': '7.5GiB'})
self.assertEqual(result.machine_type, 'n1-standard-8')
self.assertEqual(result.cpus, None)
self.assertEqual(result.memory, None)
def testMissingGpuType(self):
flags = mock_flags.MockFlags()
with self.assertRaises(errors.Config.MissingOption) as cm:
gce_virtual_machine.GceVmSpec('test_vm_spec.GCP',
flags,
machine_type='test_machine_type',
gpu_count=1)
self.assertEqual(str(cm.exception),
('gpu_type must be specified if gpu_count is set'))
def testInvalidGpuCount(self):
flags = mock_flags.MockFlags()
with self.assertRaises(errors.Config.InvalidValue) as cm:
virtual_machine.BaseVmSpec(_COMPONENT,
flag_values=flags,
gpu_count=0,
gpu_type='k80')
self.assertEqual(str(cm.exception),
('Invalid test_component.gpu_count value: "0". '
'Value must be at least 1.'))
def testInvalidGpuCount(self):
flags = mock_flags.MockFlags()
with self.assertRaises(errors.Config.InvalidValue) as cm:
gce_virtual_machine.GceVmSpec('test_vm_spec.GCP',
flags,
machine_type='test_machine_type',
gpu_count=0,
gpu_type='k80')
self.assertEqual(str(cm.exception),
('Invalid test_vm_spec.GCP.gpu_count value: "0". '
'Value must be at least 1.'))
def testInvalidGpuType(self):
flags = mock_flags.MockFlags()
with self.assertRaises(errors.Config.InvalidValue) as cm:
gce_virtual_machine.GceVmSpec('test_vm_spec.GCP',
flags,
machine_type='test_machine_type',
gpu_count=1,
gpu_type='bad_type')
self.assertEqual(
str(cm.exception),
('Invalid test_vm_spec.GCP.gpu_type value: "bad_type". '
'Value must be one of the following: k80.'))
def testInvalidGpuType(self):
flags = mock_flags.MockFlags()
with self.assertRaises(errors.Config.InvalidValue) as cm:
virtual_machine.BaseVmSpec(_COMPONENT,
flag_values=flags,
gpu_count=1,
gpu_type='bad_type')
self.assertIn(('Invalid test_component.gpu_type value: "bad_type". '
'Value must be one of the following:'),
str(cm.exception))
self.assertIn('k80', str(cm.exception))
self.assertIn('p100', str(cm.exception))
