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
)
