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))