def test_basic_footprint_w_save(self):
output_mask = df_to_px_mask(
os.path.join(data_dir, 'sample.csv'),
geom_col='PolygonWKT_Pix',
reference_im=os.path.join(data_dir, "sample_geotiff.tif"),
out_file=os.path.join(data_dir, 'test_out.tif'))
truth_mask = skimage.io.imread(
os.path.join(data_dir, 'sample_fp_from_df2px.tif'))
saved_output_mask = skimage.io.imread(
os.path.join(data_dir, 'test_out.tif'))
assert np.array_equal(output_mask, truth_mask)
assert np.array_equal(saved_output_mask, truth_mask[:, :, 0])
os.remove(os.path.join(data_dir, 'test_out.tif')) # clean up after
def test_all_three_w_save(self):
"""Test creating a 3-channel mask."""
output_mask = df_to_px_mask(
os.path.join(data_dir, 'sample.csv'),
channels=['footprint', 'boundary', 'contact'],
boundary_type='outer',
boundary_width=5,
contact_spacing=15,
geom_col='PolygonWKT_Pix',
reference_im=os.path.join(data_dir, "sample_geotiff.tif"),
out_file=os.path.join(data_dir, 'test_out.tif'))
truth_mask = skimage.io.imread(
os.path.join(data_dir, 'sample_fbc_from_df2px.tif'))
saved_output_mask = skimage.io.imread(
os.path.join(data_dir, 'test_out.tif'))
assert np.array_equal(output_mask, truth_mask)
assert np.array_equal(saved_output_mask, truth_mask)
os.remove(os.path.join(data_dir, 'test_out.tif')) # clean up after