def test_centroid_pass(self): """Read a centroid excel file correctly.""" centroids = Centroids.from_excel(HAZ_TEMPLATE_XLS) n_centroids = 45 self.assertEqual(centroids.coord.shape[0], n_centroids) self.assertEqual(centroids.coord.shape[1], 2) self.assertEqual(centroids.coord[0][0], -25.95) self.assertEqual(centroids.coord[0][1], 32.57) self.assertEqual(centroids.coord[n_centroids - 1][0], -24.7) self.assertEqual(centroids.coord[n_centroids - 1][1], 33.88)
def test_read_with_cent(self): """Test from_footprints while passing in a Centroids object""" var_names = copy.deepcopy(DEF_VAR_EXCEL) var_names['sheet_name'] = 'fp_centroids-test' var_names['col_name']['region_id'] = 'iso_n3' test_centroids = Centroids.from_excel( DATA_DIR.joinpath('fp_centroids-test.xls'), var_names=var_names ) storms = StormEurope.from_footprints(WS_DEMO_NC, centroids=test_centroids) self.assertEqual(storms.intensity.shape, (2, 9944)) self.assertEqual( np.count_nonzero( ~np.isnan(storms.centroids.region_id) ), 6401 )