def test_cog_file(tmpdir):
pp = Path(str(tmpdir))
xx, ds = gen_test_data(pp)
# write to file
ff = write_cog(xx, pp / "cog.tif")
assert isinstance(ff, Path)
assert ff == pp / "cog.tif"
assert ff.exists()
yy = rio_slurp_xarray(pp / "cog.tif")
np.testing.assert_array_equal(yy.values, xx.values)
assert yy.geobox == xx.geobox
assert yy.nodata == xx.nodata
_write_cog(np.stack([xx.values, xx.values]),
xx.geobox,
pp / "cog-2-bands.tif",
overview_levels=[])
yy, mm = rio_slurp(pp / "cog-2-bands.tif")
assert mm.gbox == xx.geobox
assert yy.shape == (2, *xx.shape)
np.testing.assert_array_equal(yy[0], xx.values)
np.testing.assert_array_equal(yy[1], xx.values)
with pytest.raises(ValueError, match="Need 2d or 3d ndarray on input"):
_write_cog(xx.values.ravel(), xx.geobox, pp / "wontwrite.tif")
def test_cog_rgba(tmpdir, use_windowed_writes):
pp = Path(str(tmpdir))
xx, ds = gen_test_data(pp)
pix = np.dstack([xx.values] * 4)
rgba = xr.DataArray(pix,
attrs=xx.attrs,
dims=("y", "x", "band"),
coords=xx.coords)
assert rgba.geobox == xx.geobox
assert rgba.shape[:2] == rgba.geobox.shape
ff = write_cog(rgba,
pp / "cog.tif",
use_windowed_writes=use_windowed_writes)
yy = rio_slurp_xarray(ff)
assert yy.geobox == rgba.geobox
assert yy.shape == rgba.shape
np.testing.assert_array_equal(yy.values, rgba.values)
with pytest.raises(ValueError):
_write_cog(
rgba.values[1:, :, :],
rgba.geobox,
":mem:",
use_windowed_writes=use_windowed_writes,
)
def test_cog_file(tmpdir, opts):
pp = Path(str(tmpdir))
xx, ds = gen_test_data(pp)
# write to file
ff = write_cog(xx, pp / "cog.tif", **opts)
assert isinstance(ff, Path)
assert ff == pp / "cog.tif"
assert ff.exists()
yy = rio_slurp_xarray(pp / "cog.tif")
np.testing.assert_array_equal(yy.values, xx.values)
assert yy.geobox == xx.geobox
assert yy.nodata == xx.nodata
_write_cog(np.stack([xx.values, xx.values]),
xx.geobox,
pp / "cog-2-bands.tif",
overview_levels=[],
**opts)
yy, mm = rio_slurp(pp / "cog-2-bands.tif")
assert mm.gbox == xx.geobox
assert yy.shape == (2, *xx.shape)
np.testing.assert_array_equal(yy[0], xx.values)
np.testing.assert_array_equal(yy[1], xx.values)
with pytest.raises(ValueError, match="Need 2d or 3d ndarray on input"):
_write_cog(xx.values.ravel(), xx.geobox, pp / "wontwrite.tif")
# sizes that are not multiples of 16
# also check that supplying `nodata=` doesn't break things
xx_odd = xx[:23, :63]
ff = write_cog(xx_odd,
pp / "cog_odd.tif",
nodata=xx_odd.attrs["nodata"],
**opts)
assert isinstance(ff, Path)
assert ff == pp / "cog_odd.tif"
assert ff.exists()
yy = rio_slurp_xarray(pp / "cog_odd.tif")
np.testing.assert_array_equal(yy.values, xx_odd.values)
assert yy.geobox == xx_odd.geobox
assert yy.nodata == xx_odd.nodata
with pytest.warns(UserWarning):
write_cog(xx, pp / "cog_badblocksize.tif", blocksize=50)
def test_cog_rgba(tmpdir):
pp = Path(str(tmpdir))
xx, ds = gen_test_data(pp)
pix = np.dstack([xx.values] * 4)
rgba = xr.DataArray(pix,
attrs=xx.attrs,
dims=('y', 'x', 'band'),
coords=xx.coords)
assert(rgba.geobox == xx.geobox)
assert(rgba.shape[:2] == rgba.geobox.shape)
ff = write_cog(rgba, pp / "cog.tif")
yy = rio_slurp_xarray(ff)
assert(yy.geobox == rgba.geobox)
assert(yy.shape == rgba.shape)
np.testing.assert_array_equal(yy.values, rgba.values)
with pytest.raises(ValueError):
_write_cog(rgba.values[1:, :, :], rgba.geobox, ':mem:')