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))