Skip to content

Commit 58b60b5

Browse files
update tests
1 parent 10e5e28 commit 58b60b5

File tree

2 files changed

+16
-4
lines changed

2 files changed

+16
-4
lines changed

climada/hazard/tc_tracks.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1703,7 +1703,7 @@ def from_FAST(cls, folder_name: str):
17031703
).astype("datetime64[s]")
17041704

17051705
# Define variables
1706-
max_sustained_wind_knots = track.v_trks.data * 1.943844
1706+
max_sustained_wind_knots = track.vmax_trks.data * 1.943844
17071707
env_pressure = BASIN_ENV_PRESSURE[track.tc_basins.data.item()]
17081708
cen_pres = _estimate_pressure(
17091709
np.full(lat.shape, np.nan), lat, lon, max_sustained_wind_knots
@@ -1716,7 +1716,7 @@ def from_FAST(cls, folder_name: str):
17161716
"time",
17171717
np.full(time.shape[0], track.time.data[1]),
17181718
),
1719-
"max_sustained_wind": ("time", track.v_trks.data),
1719+
"max_sustained_wind": ("time", track.vmax_trks.data),
17201720
"central_pressure": ("time", cen_pres),
17211721
"radius_max_wind": (
17221722
"time",

climada/hazard/test/test_tc_tracks.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -658,7 +658,7 @@ def test_from_FAST(self):
658658
"orig_event_flag": True,
659659
"data_provider": "FAST",
660660
"id_no": 0,
661-
"category": 0,
661+
"category": 1,
662662
}
663663

664664
self.assertIsInstance(
@@ -674,8 +674,20 @@ def test_from_FAST(self):
674674
)
675675
self.assertEqual(len(tc_track.data), 5)
676676
self.assertEqual(tc_track.data[0].attrs, expected_attributes)
677-
self.assertEqual(tc_track.data[0].environmental_pressure.data[0], 1010)
678677
self.assertEqual(list(tc_track.data[0].coords.keys()), ["time", "lat", "lon"])
678+
self.assertEqual(
679+
tc_track.data[0].time.values[0],
680+
np.datetime64("2025-09-01T00:00:00.000000000"),
681+
)
682+
self.assertEqual(tc_track.data[0].lat.values[0], 17.863591350508266)
683+
self.assertEqual(tc_track.data[0].lon.values[0], 288.2355824168037)
684+
self.assertEqual(len(tc_track.data[0].time), 121)
685+
self.assertEqual(tc_track.data[0].time_step[0], 10800)
686+
self.assertEqual(
687+
tc_track.data[0].max_sustained_wind.values[10], 24.71636959089841
688+
)
689+
self.assertEqual(tc_track.data[0].environmental_pressure.data[0], 1010)
690+
self.assertEqual(tc_track.data[0].basin[0], "NA")
679691

680692
def test_to_geodataframe_points(self):
681693
"""Conversion of TCTracks to GeoDataFrame using Points."""

0 commit comments

Comments
 (0)