Update test snapshot

bieniu · Apr 23, 2024 · 5151257 · 5151257
1 parent 0814ffb
commit 5151257
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/imgw_pib/model.py b/imgw_pib/model.py
@@ -30,8 +30,10 @@ class WeatherData(ImgwPibData):
     wind_speed: SensorData
     wind_direction: SensorData
     precipitation: SensorData
+
     station: str
     station_id: str
+
     measurement_date: datetime | None
 
 
@@ -40,14 +42,17 @@ class HydrologicalData(ImgwPibData):
     """Hudrological Data class for IMGW-PIB."""
 
     water_level: SensorData
-    warning_water_level: SensorData
     alarm_water_level: SensorData
+    warning_water_level: SensorData
     water_temperature: SensorData
-    station: str
+
     river: str
     station_id: str
+    station: str
+
     water_level_measurement_date: datetime | None
     water_temperature_measurement_date: datetime | None
+
     alarm_level: bool | None = None
     warning_level: bool | None = None
 

diff --git a/tests/snapshots/test_init.ambr b/tests/snapshots/test_init.ambr
@@ -1,6 +1,6 @@
 # serializer version: 1
 # name: test_hydrological_station
-  HydrologicalData(water_level=SensorData(name='Water Level', value=568.0, unit='cm'), warning_water_level=SensorData(name='Warning Water Level', value=590.0, unit='cm'), alarm_water_level=SensorData(name='Alarm Water Level', value=630.0, unit='cm'), water_temperature=SensorData(name='Water Temperature', value=8.1, unit='°C'), station='Nowe Batorowo', river='Zalew Wiślany', station_id='154190050', water_level_measurement_date=datetime.datetime(2024, 4, 22, 10, 0, tzinfo=datetime.timezone.utc), water_temperature_measurement_date=datetime.datetime(2024, 4, 22, 10, 10, tzinfo=datetime.timezone.utc))
+  HydrologicalData(water_level=SensorData(name='Water Level', value=568.0, unit='cm'), alarm_water_level=SensorData(name='Alarm Water Level', value=630.0, unit='cm'), warning_water_level=SensorData(name='Warning Water Level', value=590.0, unit='cm'), water_temperature=SensorData(name='Water Temperature', value=8.1, unit='°C'), river='Zalew Wiślany', station_id='154190050', station='Nowe Batorowo', water_level_measurement_date=datetime.datetime(2024, 4, 22, 10, 0, tzinfo=datetime.timezone.utc), water_temperature_measurement_date=datetime.datetime(2024, 4, 22, 10, 10, tzinfo=datetime.timezone.utc), alarm_level=False, warning_level=False)
 # ---
 # name: test_hydrological_stations
   dict({