add test

Author: NicolasColombi

climada/entity/impact_funcs/test/test_tc.py

@@ -168,6 +168,46 @@ def test_get_countries_per_region(self):
         self.assertListEqual(out[2], [124, 840])
         self.assertListEqual(out[3], ["CAN", "USA"])
 
+    def test_get_region_per_countries(self):
+        """Test static get_regions_per_countries()"""
+        ifs = ImpfSetTropCyclone()
+        out = ifs.get_regions_per_countries(countries=["CHE"], code_type="ISO3A")
+        self.assertEqual(out[0][0], 10)
+        self.assertEqual(out[1][0], "ROW")
+        self.assertEqual(out[2][0], "Rest of The World")
+        out = ifs.get_regions_per_countries(countries=[756], code_type="ISO3N")
+        self.assertEqual(out[0][0], 10)
+        self.assertEqual(out[1][0], "ROW")
+        self.assertEqual(out[2][0], "Rest of The World")
+        with self.assertRaises(ValueError) as context:
+            ImpfSetTropCyclone.get_regions_per_countries(
+                countries=["SWE"], code_type="invalid_code"
+            )
+        self.assertEqual(
+            str(context.exception), "code_type must be either 'iso3a' or 'iso3n'"
+        )
+        with self.assertRaises(ValueError) as context:
+            ImpfSetTropCyclone.get_regions_per_countries(
+                countries=["SWE", 840], code_type="ISO3A"
+            )
+        self.assertEqual(
+            str(context.exception), "All elements in the list must be of the same type."
+        )
+        with self.assertRaises(ValueError) as context:
+            ImpfSetTropCyclone.get_regions_per_countries(
+                countries=[840, 124], code_type="ISO3A"
+            )
+        self.assertEqual(
+            str(context.exception), "ISO3A code type cannot have integer values."
+        )
+        with self.assertRaises(ValueError) as context:
+            ImpfSetTropCyclone.get_regions_per_countries(
+                countries=["MEX"], code_type="ISO3N"
+            )
+        self.assertEqual(
+            str(context.exception), "ISO3N code type cannot have string values."
+        )
+
 
 # Execute Tests
 if __name__ == "__main__":

climada/entity/impact_funcs/trop_cyclone.py

@@ -480,6 +480,12 @@ def get_regions_per_countries(
 
         if code_type not in {"ISO3A", "ISO3N"}:
             raise ValueError("code_type must be either 'iso3a' or 'iso3n'")
+        elif not all(isinstance(country, type(countries[0])) for country in countries):
+            raise ValueError("All elements in the list must be of the same type.")
+        elif code_type == "ISO3A" and isinstance((countries[0]), int):
+            raise ValueError("ISO3A code type cannot have integer values.")
+        elif code_type == "ISO3N" and isinstance((countries[0]), str):
+            raise ValueError("ISO3N code type cannot have string values.")
 
         country_dict = getattr(CountryCode, code_type).value
         # Find region