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)
