@@ -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