Update unit tests to check the validation works

Author: JamieTaylor-TUOS

Tests/test_pvlive_api.py

@@ -5,11 +5,12 @@
 """
 
 import unittest
+from unittest.mock import MagicMock
 from datetime import datetime, date, time
 import pytz
 
 import pandas.api.types as ptypes
-from pvlive_api import PVLive
+from pvlive_api import PVLive, PVLiveException
 
 class PVLiveTestCase(unittest.TestCase):
     """Tests for `pvlive.py`."""
@@ -139,6 +140,9 @@ def test_latest(self):
         data = self.api.latest(entity_type="gsp", entity_id=103, dataframe=True)
         self.check_df_columns(data)
         self.check_df_dtypes(data)
+        self.api._fetch_url = MagicMock(return_value={"notdata": [], "notmeta": []})
+        with self.assertRaises(PVLiveException):
+            data = self.api.latest(entity_type="gsp", entity_id=0, dataframe=True)
 
     def test_day_peak(self):
         """Tests the day_peak function."""

pvlive_api/__init__.py

@@ -1,3 +1,3 @@
-from pvlive_api.pvlive import PVLive
+from pvlive_api.pvlive import PVLive, PVLiveException
 
-__all__ = ["PVLive"]
+__all__ = ["PVLive", "PVLiveException"]