def test_mat_pass(self): """Read a centroid mat file correctly.""" centroids = Centroids.from_mat(HAZ_TEST_MAT) n_centroids = 100 self.assertEqual(centroids.coord.shape, (n_centroids, 2)) self.assertEqual(centroids.coord[0][0], 21) self.assertEqual(centroids.coord[0][1], -84) self.assertEqual(centroids.coord[n_centroids - 1][0], 30) self.assertEqual(centroids.coord[n_centroids - 1][1], -75)
from climada.hazard.trop_cyclone import (TropCyclone, _close_centroids, _vtrans, _B_holland_1980, _bs_holland_2008, _v_max_s_holland_2008, _x_holland_2010, _stat_holland_1980, _stat_holland_2010) from climada.hazard.centroids.centr import Centroids import climada.hazard.test as hazard_test DATA_DIR = Path(hazard_test.__file__).parent.joinpath('data') HAZ_TEST_MAT = DATA_DIR.joinpath('atl_prob_no_name.mat') TEST_TRACK = DATA_DIR.joinpath("trac_brb_test.csv") TEST_TRACK_SHORT = DATA_DIR.joinpath("trac_short_test.csv") CENTR_TEST_BRB = Centroids.from_mat(DATA_DIR.joinpath('centr_brb_test.mat')) class TestReader(unittest.TestCase): """Test loading funcions from the TropCyclone class""" def test_set_one_pass(self): """Test _tc_from_track function.""" intensity_idx = [0, 1, 2, 3, 80, 100, 120, 200, 220, 250, 260, 295] intensity_values = { "geosphere": [ 25.60794285, 26.90906280, 28.26649026, 25.54076797, 31.21986961, 36.17171808, 21.11408573, 28.01457948, 32.65349378, 31.34027741, 0, 40.27362679 ], "equirect": [ 25.60778909, 26.90887264, 28.26624642, 25.54092386,
def test_mat_global_pass(self): """Test read GLB_CENTROIDS_MAT""" centroids = Centroids.from_mat(GLB_CENTROIDS_MAT) self.assertEqual(centroids.region_id[1062443], 35) self.assertEqual(centroids.region_id[170825], 28)