Skip to content

Commit bb0db52

Browse files
remove numeric from enum, convert numeric to alpha3
1 parent f6ceac0 commit bb0db52

File tree

2 files changed

+270
-108
lines changed

2 files changed

+270
-108
lines changed

climada/entity/impact_funcs/test/test_tc.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def test_default_values_pass(self):
115115
self.assertEqual(impfs.size(), 10)
116116
self.assertEqual(impfs.get_ids()["TC"], [1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
117117
self.assertEqual(impf_wp4.intensity_unit, "m/s")
118-
self.assertEqual(impf_wp4.name, "North West Pacific (WP4)")
118+
self.assertEqual(impf_wp4.name, "North West Pacific")
119119
self.assertAlmostEqual(v_halfs["WP2"], 188.4, places=7)
120120
self.assertAlmostEqual(v_halfs["ROW"], 110.1, places=7)
121121
self.assertListEqual(list(impf_wp4.intensity), list(np.arange(0, 121, 5)))
@@ -185,12 +185,17 @@ def test_get_imf_id_regions_per_countries(self):
185185
self.assertEqual(impf_id_reg_id_reg_name[1][0], "ROW")
186186
self.assertEqual(impf_id_reg_id_reg_name[2][0], "Rest of The World")
187187

188-
with self.assertRaises(ValueError) as context:
189-
ImpfSetTropCyclone.get_impf_id_regions_per_countries(countries=[3.22])
190-
self.assertEqual(
191-
str(context.exception),
192-
"The element 3.22 is neither in ISO3A nor ISO3N format",
188+
impf_id_reg_id_reg_name = ifs.get_impf_id_regions_per_countries(
189+
countries=["CHE", 268]
193190
)
191+
# CHE
192+
self.assertEqual(impf_id_reg_id_reg_name[0][0], 10)
193+
self.assertEqual(impf_id_reg_id_reg_name[1][0], "ROW")
194+
self.assertEqual(impf_id_reg_id_reg_name[2][0], "Rest of The World")
195+
# GEO (georgia, 268)
196+
self.assertEqual(impf_id_reg_id_reg_name[0][1], 3)
197+
self.assertEqual(impf_id_reg_id_reg_name[1][1], "NI")
198+
self.assertEqual(impf_id_reg_id_reg_name[2][1], "North Indian")
194199

195200

196201
# Execute Tests

0 commit comments

Comments
 (0)