def test_extract_roi_1(training_raster, training_roi):
dat, label, xs, ys = next(roi.extract_roi(training_raster,
training_roi,
feature_prop='code_1',
all_touched=False))
geom = shapely.geometry.shape(training_roi[0]['geometry'])
x_size, y_size = training_raster.res
xy = np.stack((xs + int(x_size / 2), ys - int(y_size / 2)))
contains = []
for _xy in xy.T:
p = shapely.geometry.Point(_xy)
contains.append(geom.contains(p))
assert sum(contains) / float(len(contains)) == 1.0
def test_extract_roi_1(training_raster, training_roi):
dat, label, xs, ys = next(
roi.extract_roi(training_raster,
training_roi,
feature_prop='code_1',
all_touched=False))
geom = shapely.geometry.shape(training_roi[0]['geometry'])
x_size, y_size = training_raster.res
xy = np.stack((xs + int(x_size / 2), ys - int(y_size / 2)))
contains = []
for _xy in xy.T:
p = shapely.geometry.Point(_xy)
contains.append(geom.contains(p))
assert sum(contains) / float(len(contains)) == 1.0
def test_extract_roi_2(training_raster, training_roi):
# Test with ds.nodata included
meta = training_raster.meta
meta['nodata'] = 255
meta['driver'] = 'MEM'
with rasterio.open('tmp', 'w', **meta) as ds:
dat, label, xs, ys = next(roi.extract_roi(ds,
training_roi,
feature_prop='code_1',
all_touched=False))
geom = shapely.geometry.shape(training_roi[0]['geometry'])
x_size, y_size = training_raster.res
xy = np.stack((xs + int(x_size / 2), ys - int(y_size / 2)))
contains = []
for _xy in xy.T:
p = shapely.geometry.Point(_xy)
contains.append(geom.contains(p))
assert sum(contains) / float(len(contains)) == 1.0
assert dat.sum() == 0
def test_extract_roi_2(training_raster, training_roi):
# Test with ds.nodata included
meta = training_raster.meta
meta['nodata'] = 255
meta['driver'] = 'MEM'
with rasterio.open('tmp', 'w', **meta) as ds:
dat, label, xs, ys = next(
roi.extract_roi(ds,
training_roi,
feature_prop='code_1',
all_touched=False))
geom = shapely.geometry.shape(training_roi[0]['geometry'])
x_size, y_size = training_raster.res
xy = np.stack((xs + int(x_size / 2), ys - int(y_size / 2)))
contains = []
for _xy in xy.T:
p = shapely.geometry.Point(_xy)
contains.append(geom.contains(p))
assert sum(contains) / float(len(contains)) == 1.0
assert dat.sum() == 0
