test: cover input validation and daily_max=False

Author: markus-kreft

tests/baseline_images/test_daily_max_false.png

@@ -295,6 +295,40 @@ def test_empty_series(self):
             heatmapfigure(series, rasterized=True)
         self.assertEqual(str(cm.exception), "Series cannot be empty")
 
+    @compare_to_baseline
+    def test_daily_max_false(self):
+        # daily_max=False omits the daily peak scatter (and the twinx axis)
+        date_range = pd.date_range(
+            "2020-01-01", "2020-12-31 23:59:59", freq="15min", tz="UTC"
+        )
+        data = date_range.to_series().apply(sinus_pattern)
+        series = pd.Series(index=date_range, data=data)
+        return heatmapfigure(series, rasterized=True, daily_max=False)
+
+    def test_non_series_raises(self):
+        with self.assertRaises(TypeError) as cm:
+            heatmapfigure([0, 1, 2])  # ty: ignore[invalid-argument-type]
+        self.assertEqual(str(cm.exception), "series must be a pandas Series")
+
+    def test_non_datetime_index_raises(self):
+        series = pd.Series([0, 1, 2], index=[0, 1, 2])
+        with self.assertRaisesRegex(ValueError, "DatetimeIndex"):
+            heatmapfigure(series, rasterized=True)
+
+    def test_missing_freq_raises(self):
+        index = pd.DatetimeIndex(["2020-01-01", "2020-01-03", "2020-01-04"], tz="UTC")
+        series = pd.Series([0, 1, 2], index=index)
+        with self.assertRaisesRegex(ValueError, "frequency"):
+            heatmapfigure(series, rasterized=True)
+
+    def test_invalid_daily_func_raises(self):
+        date_range = pd.date_range(
+            "2020-01-01", "2020-12-31 23:59:59", freq="15min", tz="UTC"
+        )
+        series = pd.Series(index=date_range, data=[0, 1] * int(len(date_range) / 2))
+        with self.assertRaisesRegex(ValueError, "daily_func"):
+            heatmapfigure(series, rasterized=True, daily_func="median")
+
 
 class TestSuntimes(unittest.TestCase):
     """Test suite for sunrise and sunset time calculations."""