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