def test_make_binary_and_fp(self):
"""test creating a boundary mask and a fp mask together."""
output_mask = boundary_mask(df=os.path.join(data_dir, 'sample.csv'),
geom_col="PolygonWKT_Pix")
truth_mask = skimage.io.imread(
os.path.join(data_dir, 'sample_b_mask_inner.tif'))
assert np.array_equal(output_mask, truth_mask)
def test_make_outer_mask_from_fp(self):
"""test creating a boundary mask using an existing footprint mask."""
fp_mask = skimage.io.imread(
os.path.join(data_dir, 'sample_fp_mask.tif'))
output_mask = boundary_mask(fp_mask, boundary_type="outer")
truth_mask = skimage.io.imread(
os.path.join(data_dir, 'sample_b_mask_outer.tif'))
assert np.array_equal(output_mask, truth_mask)
def test_make_thick_outer_mask_from_fp(self):
"""test creating a 10-px thick boundary mask."""
fp_mask = skimage.io.imread(
os.path.join(data_dir, 'sample_fp_mask.tif'))
output_mask = boundary_mask(fp_mask,
boundary_type="outer",
boundary_width=10)
truth_mask = skimage.io.imread(
os.path.join(data_dir, 'sample_b_mask_outer_10.tif'))
assert np.array_equal(output_mask, truth_mask)