From 7b4c4a49ac78bb913fb7f15594fcb702357dca0b Mon Sep 17 00:00:00 2001 From: Nitin Garg Date: Tue, 17 Sep 2024 07:26:12 +0000 Subject: [PATCH] add utils unit tests for resource limits --- .../examples/utils/utils_test.py | 42 +++++++++++++++++++ 1 file changed, 42 insertions(+) diff --git a/perfmetrics/scripts/testing_on_gke/examples/utils/utils_test.py b/perfmetrics/scripts/testing_on_gke/examples/utils/utils_test.py index 54d400ef68..dae89cd41a 100644 --- a/perfmetrics/scripts/testing_on_gke/examples/utils/utils_test.py +++ b/perfmetrics/scripts/testing_on_gke/examples/utils/utils_test.py @@ -88,6 +88,48 @@ def test_timestamp_to_epoch_with_nznano(self): timestamp_to_epoch('2024-08-21T19:20:25.547456'), 1724268025 ) + 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']) + if __name__ == '__main__': unittest.main()