Address code review feedback

- Move math import to top of test file
- Use np_array instead of query_values for percentile calculation for consistency

Co-authored-by: 4n4nd <22333506+4n4nd@users.noreply.github.com>

Author: Copilot

prometheus_api_client/prometheus_connect.py

@@ -612,7 +612,7 @@ def get_metric_aggregation(
             elif operation.startswith("percentile"):
                 percentile = float(operation.split("_")[1])
                 aggregated_values["percentile_" + str(percentile)] = numpy.nanpercentile(
-                    query_values, percentile
+                    np_array, percentile
                 )
             elif operation == "deviation":
                 aggregated_values["deviation"] = numpy.nanstd(np_array)

tests/test_prometheus_connect.py

@@ -1,5 +1,6 @@
 """Test module for class PrometheusConnect."""
 
+import math
 import unittest
 import os
 from datetime import datetime, timedelta
@@ -441,8 +442,6 @@ def test_get_metric_aggregation_with_nan_values(self):  # noqa D102
             self.assertIn("percentile_50.0", aggregated_values)
 
             # Verify that results are not NaN
-            import math
-
             self.assertFalse(math.isnan(aggregated_values["sum"]), "Sum should not be NaN")
             self.assertFalse(math.isnan(aggregated_values["max"]), "Max should not be NaN")
             self.assertFalse(math.isnan(aggregated_values["min"]), "Min should not be NaN")