From 7cd1bb66fd552ae41c891220740e77984d558a5b Mon Sep 17 00:00:00 2001
From: Anders Nord <dr.anders.nord@gmail.com>
Date: Wed, 24 Apr 2024 07:02:49 +0200
Subject: [PATCH] test _find_missing_data

---
 tests/unit/test_unit_smhi.py | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/tests/unit/test_unit_smhi.py b/tests/unit/test_unit_smhi.py
index 3275e91..ad064b3 100644
--- a/tests/unit/test_unit_smhi.py
+++ b/tests/unit/test_unit_smhi.py
@@ -188,7 +188,7 @@ def test_interpolate(
     def test_iterate_over_time(
         self,
     ):
-        """Unit test for SMHI _interpolate method.
+        """Unit test for SMHI _iterate_over_time method.
 
         Args:
         """
@@ -220,3 +220,29 @@ def test_iterate_over_time(
 
         data = client._iterate_over_time(df, nearby_df, missing_df)
         assert data.tail(2).iloc[0]["Temperatur"] == nearby_df.iloc[0]["Temperatur"]
+
+    def test_find_missing_data(
+        self,
+    ):
+        """Unit test for SMHI _find_missing_data method.
+
+        Args:
+        """
+        df = pd.DataFrame(
+            {
+                "date": [
+                    "2024-04-21 10:00",
+                    "2024-04-21 11:00",
+                    "2024-04-21 12:00",
+                    "2024-04-22 12:00",
+                ],
+                "Temperatur": [1, 1, 1, 12],
+            }
+        )
+        df["date"] = pd.to_datetime(df["date"])
+        df = df.set_index("date")
+
+        client = SMHI()
+
+        missingdata = client._find_missing_data(df)
+        assert missingdata.iloc[0]["Temperatur"] == df.iloc[-1]["Temperatur"]