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 1c4d2b3b57..0c5b07f8c8 100644 --- a/perfmetrics/scripts/testing_on_gke/examples/utils/utils_test.py +++ b/perfmetrics/scripts/testing_on_gke/examples/utils/utils_test.py @@ -35,7 +35,7 @@ def setUpClass(self): self.end = '2024-09-25 07:06:22 UTC' def test_get_memory(self): - low1, high1 = get_memory_from_monitoring_api( + low, high = get_memory_from_monitoring_api( project_id=self.project_id, cluster_name=self.cluster_name, pod_name=self.pod_name, @@ -43,11 +43,11 @@ def test_get_memory(self): start_epoch=self.start_epoch, end_epoch=self.end_epoch, ) - self.assertLessEqual(low1, high1) - self.assertGreater(high1, 0) + self.assertLessEqual(low, high) + self.assertGreater(high, 0) def test_get_cpu(self): - low1, high1 = get_cpu_from_monitoring_api( + low, high = get_cpu_from_monitoring_api( project_id=self.project_id, cluster_name=self.cluster_name, pod_name=self.pod_name, @@ -55,8 +55,8 @@ def test_get_cpu(self): start_epoch=self.start_epoch, end_epoch=self.end_epoch, ) - self.assertLessEqual(low1, high1) - self.assertGreater(high1, 0) + self.assertLessEqual(low, high) + self.assertGreater(high, 0) def test_timestamp_to_epoch(self): self.assertEqual(timestamp_to_epoch('2024-08-21T19:20:25'), 1724268025)