Test made to accept 'us' or 'ns' precision. Fixes #185

orion-junkins · Feb 14, 2024 · 8507d6f · 8507d6f
1 parent f5b9c6b
commit 8507d6f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/forecasting/data_fetching_utilities/weather_provider/test_aws_weather_provider.py b/tests/forecasting/data_fetching_utilities/weather_provider/test_aws_weather_provider.py
@@ -66,7 +66,7 @@ def test_fetch_historical(weather_provider):
     weather_datums = weather_provider.fetch_historical()
     assert len(weather_datums) == len(weather_provider.coordinates)
     for weather_datum in weather_datums:
-        assert weather_datum.hourly_parameters.index.dtype == "datetime64[ns, UTC]"
+        assert (weather_datum.hourly_parameters.index.dtype == "datetime64[ns, UTC]" or weather_datum.hourly_parameters.index.dtype == "datetime64[us, UTC]")
         assert expected_columns == sorted(list(weather_datum.hourly_parameters.columns))
 
 
@@ -113,7 +113,7 @@ def test_fetch_current(weather_provider):
     weather_datums = weather_provider.fetch_current()
     assert len(weather_datums) == len(weather_provider.coordinates)
     for weather_datum in weather_datums:
-        assert weather_datum.hourly_parameters.index.dtype == "datetime64[ns, UTC]"
+        assert (weather_datum.hourly_parameters.index.dtype == "datetime64[ns, UTC]" or weather_datum.hourly_parameters.index.dtype == "datetime64[us, UTC]")
         assert expected_columns == sorted(list(weather_datum.hourly_parameters.columns))