def test_elevation_3_pass(self): """ Test set_elevation """ centr_ras = Centroids() centr_ras.set_raster_file(HAZ_DEMO_FL, window=Window(0, 0, 10, 20)) centr_ras.set_elevation(product='SRTM3', resampling=Resampling.nearest) self.assertEqual(centr_ras.elevation.max(), 1052) self.assertEqual(centr_ras.elevation.min(), 357) self.assertEqual(centr_ras.elevation[25], 476) self.assertEqual(centr_ras.elevation[32], 471) self.assertEqual(centr_ras.elevation[0], 695) self.assertEqual(centr_ras.elevation[-1], 661)
def test_elevation_3_pass(self): """ test set_elevation """ centr = Centroids() centr.lat, centr.lon, centr.geometry = self.data_vector() centr.geometry.crs = DEF_CRS centr.set_elevation(product='SRTM3', resampling=Resampling.nearest) centr.set_on_land() self.assertTrue(np.all(centr.elevation[centr.on_land] > 0)) self.assertTrue( np.all( centr.elevation[np.logical_not(centr.on_land)] == DEM_NODATA)) self.assertEqual(centr.elevation[0], 23) self.assertEqual(centr.elevation[1], 92) self.assertEqual(centr.elevation[2], 69) self.assertEqual(centr.elevation[3], 77) self.assertEqual(centr.elevation[4], 133) self.assertEqual(centr.elevation[5], 41) self.assertEqual(centr.elevation.min(), DEM_NODATA)