Refine climate metrics and data pipeline

tests/test_gridmet_heat_index.py

@@ -0,0 +1,55 @@
+from __future__ import annotations
+
+import sys
+import unittest
+from pathlib import Path
+
+import numpy as np
+
+SCRIPTS_DIR = Path(__file__).resolve().parents[1] / "scripts"
+sys.path.insert(0, str(SCRIPTS_DIR))
+
+from summarize_county_gridmet_humidity import (  # noqa: E402
+    DEFAULT_HEAT_INDEX_THRESHOLD_F,
+    _heat_index_f,
+)
+
+
+class HeatIndexTests(unittest.TestCase):
+    def test_matches_nws_example(self) -> None:
+        result = _heat_index_f(np.array([100.0]), np.array([55.0]))
+
+        self.assertAlmostEqual(result[0], 124.0, delta=0.5)
+
+    def test_uses_simple_formula_below_regression_range(self) -> None:
+        result = _heat_index_f(np.array([70.0]), np.array([50.0]))
+
+        self.assertAlmostEqual(result[0], 69.525, places=3)
+
+    def test_default_threshold_is_extreme_caution_boundary(self) -> None:
+        heat_index = _heat_index_f(
+            np.array([85.0, 90.0]),
+            np.array([40.0, 55.0]),
+        )
+
+        self.assertEqual(DEFAULT_HEAT_INDEX_THRESHOLD_F, 90.0)
+        self.assertEqual(
+            (heat_index >= DEFAULT_HEAT_INDEX_THRESHOLD_F).tolist(),
+            [False, True],
+        )
+
+    def test_relative_humidity_is_clamped_to_physical_range(self) -> None:
+        result = _heat_index_f(
+            np.array([90.0, 90.0]),
+            np.array([-5.0, 105.0]),
+        )
+        expected = _heat_index_f(
+            np.array([90.0, 90.0]),
+            np.array([0.0, 100.0]),
+        )
+
+        np.testing.assert_allclose(result, expected)
+
+
+if __name__ == "__main__":
+    unittest.main()