mirror of
https://github.com/rasterio/rasterio.git
synced 2026-01-25 14:10:17 +00:00
111 lines
3.0 KiB
Python
111 lines
3.0 KiB
Python
import rasterio
|
|
from rasterio.rio.main import main_group
|
|
|
|
|
|
def test_stack(tmpdir, runner):
|
|
outputname = str(tmpdir.join("stacked.tif"))
|
|
result = runner.invoke(main_group, ["stack", "tests/data/RGB.byte.tif", outputname])
|
|
assert result.exit_code == 0
|
|
with rasterio.open(outputname) as out:
|
|
assert out.count == 3
|
|
assert out.read(1).max() > 0
|
|
|
|
|
|
def test_stack_2(tmpdir, runner):
|
|
outputname = str(tmpdir.join("stacked.tif"))
|
|
result = runner.invoke(
|
|
main_group,
|
|
["stack", "tests/data/RGB.byte.tif", "tests/data/RGB.byte.tif", outputname],
|
|
)
|
|
assert result.exit_code == 0
|
|
with rasterio.open(outputname) as out:
|
|
assert out.count == 6
|
|
assert out.read(1).max() > 0
|
|
|
|
|
|
def test_stack_disjoint(tmpdir, runner):
|
|
outputname = str(tmpdir.join("stacked.tif"))
|
|
result = runner.invoke(
|
|
main_group,
|
|
[
|
|
"stack",
|
|
"tests/data/rgb1.tif",
|
|
"tests/data/rgb2.tif",
|
|
"tests/data/rgb3.tif",
|
|
"tests/data/rgb4.tif",
|
|
outputname,
|
|
],
|
|
)
|
|
assert result.exit_code == 0
|
|
with rasterio.open(outputname) as out:
|
|
assert out.count == 12
|
|
assert out.shape == (718, 791)
|
|
|
|
|
|
def test_stack_list(tmpdir, runner):
|
|
outputname = str(tmpdir.join("stacked.tif"))
|
|
result = runner.invoke(
|
|
main_group, ["stack", "tests/data/RGB.byte.tif", "--bidx", "1,2,3", outputname]
|
|
)
|
|
assert result.exit_code == 0
|
|
with rasterio.open(outputname) as out:
|
|
assert out.count == 3
|
|
|
|
|
|
def test_stack_slice(tmpdir, runner):
|
|
outputname = str(tmpdir.join("stacked.tif"))
|
|
result = runner.invoke(
|
|
main_group,
|
|
[
|
|
"stack",
|
|
"tests/data/RGB.byte.tif",
|
|
"--bidx",
|
|
"..2",
|
|
"tests/data/RGB.byte.tif",
|
|
"--bidx",
|
|
"3..",
|
|
outputname,
|
|
],
|
|
)
|
|
assert result.exit_code == 0
|
|
with rasterio.open(outputname) as out:
|
|
assert out.count == 3
|
|
|
|
|
|
def test_stack_single_slice(tmpdir, runner):
|
|
outputname = str(tmpdir.join("stacked.tif"))
|
|
result = runner.invoke(
|
|
main_group,
|
|
[
|
|
"stack",
|
|
"tests/data/RGB.byte.tif",
|
|
"--bidx",
|
|
"1",
|
|
"tests/data/RGB.byte.tif",
|
|
"--bidx",
|
|
"2..",
|
|
"--rgb",
|
|
outputname,
|
|
],
|
|
)
|
|
assert result.exit_code == 0
|
|
with rasterio.open(outputname) as out:
|
|
assert out.count == 3
|
|
|
|
|
|
def test_format_jpeg(tmpdir, runner):
|
|
outputname = str(tmpdir.join("stacked.jpg"))
|
|
result = runner.invoke(
|
|
main_group, ["stack", "tests/data/RGB.byte.tif", outputname, "--format", "JPEG"]
|
|
)
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_error(tmpdir, runner):
|
|
outputname = str(tmpdir.join("stacked.tif"))
|
|
result = runner.invoke(
|
|
main_group,
|
|
["stack", "tests/data/RGB.byte.tif", outputname, "--driver", "BOGUS"],
|
|
)
|
|
assert result.exit_code == 1
|