def test_assign_raster_same_pass(self): """Test assign_centroids with raster hazard""" exp = Exposures() exp.set_from_raster(HAZ_DEMO_FL, window=Window(10, 20, 50, 60)) exp.check() haz = Hazard('FL') haz.set_raster([HAZ_DEMO_FL], window=Window(10, 20, 50, 60)) exp.assign_centroids(haz) np.testing.assert_array_equal(exp.gdf[INDICATOR_CENTR + 'FL'].values, np.arange(haz.centroids.size, dtype=int))
def test_assign_large_hazard_subset_pass(self): """Test assign_centroids with raster hazard""" exp = Exposures() exp.set_from_raster(HAZ_DEMO_FL, window=Window(10, 20, 50, 60)) exp.gdf.latitude[[0, 1]] = exp.gdf.latitude[[1, 0]] exp.gdf.longitude[[0, 1]] = exp.gdf.longitude[[1, 0]] exp.check() haz = Hazard('FL') haz.set_raster([HAZ_DEMO_FL]) haz.raster_to_vector() exp.assign_centroids(haz) assigned_centroids = haz.centroids.select( sel_cen=exp.gdf[INDICATOR_CENTR + 'FL'].values) np.testing.assert_array_equal(assigned_centroids.lat, exp.gdf.latitude) np.testing.assert_array_equal(assigned_centroids.lon, exp.gdf.longitude)
def test_read_raster_pass(self): """set_from_raster""" exp = Exposures() exp.set_from_raster(HAZ_DEMO_FL, window=Window(10, 20, 50, 60)) exp.check() self.assertTrue(u_coord.equal_crs(exp.crs, DEF_CRS)) self.assertAlmostEqual(exp.gdf['latitude'].max(), 10.248220966978932 - 0.009000000000000341 / 2) self.assertAlmostEqual( exp.gdf['latitude'].min(), 10.248220966978932 - 0.009000000000000341 / 2 - 59 * 0.009000000000000341) self.assertAlmostEqual(exp.gdf['longitude'].min(), -69.2471495969998 + 0.009000000000000341 / 2) self.assertAlmostEqual( exp.gdf['longitude'].max(), -69.2471495969998 + 0.009000000000000341 / 2 + 49 * 0.009000000000000341) self.assertEqual(len(exp.gdf), 60 * 50) self.assertAlmostEqual( exp.gdf.value.values.reshape((60, 50))[25, 12], 0.056825936)