add utils unit tests for resource limits

GoogleCloudPlatform · Sep 18, 2024 · c93fde8 · c93fde8
1 parent 0adc614
commit c93fde8
Showing 1 changed file with 43 additions and 0 deletions.
diff --git a/perfmetrics/scripts/testing_on_gke/examples/utils/utils_test.py b/perfmetrics/scripts/testing_on_gke/examples/utils/utils_test.py
@@ -64,6 +64,49 @@ def test_timestamp_to_epoch(self):
  self.assertEqual(timestamp_to_epoch(timestamp), expected_epoch)
  pass
 
+ def test_resource_limit(self):
+ inputs = [
+ {
+ 'nodeType': 'n2-standard-32',
+ 'expected_limits_cpu': 32,
+ 'expected_error': False,
+ },
+ {
+ 'nodeType': 'n2-standard-96',
+ 'expected_limits_cpu': 96,
+ 'expected_error': False,
+ },
+ {
+ 'nodeType': 'n2-standard-96',
+ 'expected_limits_cpu': 96,
+ 'expected_error': False,
+ },
+ {
+ 'nodeType': 'c3-standard-176',
+ 'expected_limits_cpu': 176,
+ 'expected_error': False,
+ },
+ {
+ 'nodeType': 'c3-standard-176-lssd',
+ 'expected_limits_cpu': 176,
+ 'expected_error': False,
+ },
+ {'nodeType': 'n2-standard-1', 'expected_error': True},
+ {'nodeType': 'unknown-machine-type', 'expected_error': True},
+ ]
+ for input in inputs:
+ self.assertEqual(dict, type(input))
+ try:
+ resource_limits = utils.resource_limits(input['nodeType'])
+ self.assertEqual(
+ input['expected_limits_cpu'],
+ resource_limits[0]['cpu'],
+ )
+ self.assertFalse(input['expected_error'])
+ except utils.UnknownMachineTypeError:
+ self.assertTrue(input['expected_error'])
+ pass
+
 
 if __name__ == '__main__':
  unittest.main()