def test_region_id_pass(self):
""" Test set_dist_coast """
centr_ras = Centroids()
centr_ras.set_raster_file(HAZ_DEMO_FL, window= Window(0, 0, 50, 60))
centr_ras.set_region_id()
self.assertEqual(centr_ras.region_id.size, centr_ras.size)
self.assertTrue(np.array_equal(np.unique(centr_ras.region_id), np.array([862])))
def test_region_id_pass(self):
"""Test set_region_id"""
centr = Centroids()
centr.lat, centr.lon, centr.geometry = self.data_vector()
centr.geometry.crs = {'init': 'epsg:4326'}
centr.set_region_id()
self.assertEqual(np.count_nonzero(centr.region_id), 6)
self.assertEqual(centr.region_id[0], 52) # 052 for barbados
def test_on_land(self):
"""Test set_on_land"""
centr = Centroids()
centr.lat, centr.lon, centr.geometry = self.data_vector()
centr.geometry.crs = {'init': 'epsg:4326'}
centr.set_on_land()
centr.set_region_id()
centr.region_id[centr.region_id > 0] = 1
self.assertTrue(
np.array_equal(centr.on_land.astype(int), centr.region_id))
