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()