fix tests

Author: peterdudfield

nowcasting_dataset/data_sources/pv/live.py

@@ -80,6 +80,8 @@ def get_pv_power_from_database(
     with db_connection.get_session() as session:
         pv_yields: List[PVYieldSQL] = get_pv_yield(session=session, start_utc=start_utc_extra)
 
+        logger.debug(f"Found {len(pv_yields)} from database")
+
         pv_yields_df = pd.DataFrame(
             [(PVYield.from_orm(pv_yield)).__dict__ for pv_yield in pv_yields]
         )
@@ -116,6 +118,8 @@ def get_pv_power_from_database(
         pv_yields_df.interpolate(limit=limit, inplace=True)
 
     # filter out the extra minutes loaded
+    logger.debug(f"{len(pv_yields_df)} of datetimes before filter on {start_utc}")
     pv_yields_df = pv_yields_df[pv_yields_df.index >= start_utc]
+    logger.debug(f"{len(pv_yields_df)} of datetimes after filter on {start_utc}")
 
     return pv_yields_df

tests/data_sources/pv/test_pv_live.py

@@ -19,12 +19,12 @@ def test_get_metadata_from_database(pv_yields_and_systems):
     assert len(meteadata) == 2
 
 
-@freeze_time("2022-01-01")
+@freeze_time("2022-01-01 05:00")
 def test_get_pv_power_from_database(pv_yields_and_systems):
     """Get pv power from database"""
     pv_power = get_pv_power_from_database(history_duration=timedelta(hours=1))
 
-    assert len(pv_power) == 72  # 6 hours at 5 mins = 6*12
+    assert len(pv_power) == 13  # 1 hours at 5 mins = 6*12
     assert len(pv_power.columns) == 2
     assert pv_power.columns[0] == "11"
     assert (
@@ -49,7 +49,7 @@ def test_get_pv_power_from_database_interpolate(pv_yields_and_systems):
     assert pv_power.isna().sum().sum() == 6  # the last 30 mins is still nans
 
 
-@freeze_time("2022-01-01")
+@freeze_time("2022-01-01 05:00")
 def test_get_example_and_batch(pv_yields_and_systems):
     """Test PVDataSource with data source from database"""
 
@@ -77,6 +77,6 @@ def test_get_example_and_batch(pv_yields_and_systems):
     assert len(pv_data_source.pv_metadata) > 0
 
     locations = pv_data_source.get_locations(pv_data_source.pv_power.index)
-    assert len(locations) == 72  # 6 hours at 5 mins
+    assert len(locations) == 7  # 30 minutes at 5 mins, inclusive
 
     _ = pv_data_source.get_example(location=locations[0])