From 5151257cfa564d6fc3d8cd2cc972c26993b03ad4 Mon Sep 17 00:00:00 2001
From: Maciej Bieniek <478555+bieniu@users.noreply.github.com>
Date: Tue, 23 Apr 2024 15:35:06 +0200
Subject: [PATCH] Update test snapshot

---
 imgw_pib/model.py              | 9 +++++++--
 tests/snapshots/test_init.ambr | 2 +-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/imgw_pib/model.py b/imgw_pib/model.py
index 71b6151..07d64ce 100644
--- 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
index 77fe9a7..4ab70ff 100644
--- 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({