| import unittest | |
| from src.climate import _summarize_archive | |
| class ClimateTests(unittest.TestCase): | |
| def test_archive_precipitation_is_average_monthly_total_not_multi_year_sum(self): | |
| data = { | |
| "daily": { | |
| "time": ["2024-01-01", "2024-01-02", "2025-01-01", "2025-01-02", "2024-02-01", "2025-02-01"], | |
| "temperature_2m_mean": [20, 22, 24, 26, 28, 30], | |
| "precipitation_sum": [1, 2, 3, 4, 10, 30], | |
| "relative_humidity_2m_mean": [70, 72, 74, 76, 78, 80], | |
| "wind_speed_10m_mean": [5, 6, 7, 8, 9, 10], | |
| "wind_direction_10m_dominant": [90, 100, 110, 120, 130, 140], | |
| } | |
| } | |
| summary = _summarize_archive(data) | |
| self.assertEqual(summary["months"][0]["precipitation_mm"], 5) | |
| self.assertEqual(summary["months"][1]["precipitation_mm"], 20) | |
| self.assertEqual(summary["total_precipitation_mm"], 25) | |
| if __name__ == "__main__": | |
| unittest.main() | |