mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
import logging
|
|
import sys
|
|
|
|
from click.testing import CliRunner
|
|
|
|
import rasterio
|
|
from rasterio.rio.main import main_group as cli
|
|
|
|
|
|
logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)
|
|
|
|
|
|
def test_err(data):
|
|
runner = CliRunner()
|
|
inputfile = str(data.join('RGB.byte.tif'))
|
|
result = runner.invoke(cli, ['overview', inputfile, '--build', 'a^2'])
|
|
assert result.exit_code == 2
|
|
assert "must match" in result.output
|
|
|
|
|
|
def test_ls_none(data):
|
|
runner = CliRunner()
|
|
inputfile = str(data.join('RGB.byte.tif'))
|
|
result = runner.invoke(cli, ['overview', inputfile, '--ls'])
|
|
assert result.exit_code == 0
|
|
expected = "Overview factors:\n Band 1: None (method: 'unknown')\n Band 2: None (method: 'unknown')\n Band 3: None (method: 'unknown')\n"
|
|
assert result.output == expected
|
|
|
|
|
|
def test_build_ls(data):
|
|
runner = CliRunner()
|
|
inputfile = str(data.join('RGB.byte.tif'))
|
|
result = runner.invoke(cli, ['overview', inputfile, '--build', '2,4,8'])
|
|
assert result.exit_code == 0
|
|
result = runner.invoke(cli, ['overview', inputfile, '--ls'])
|
|
assert result.exit_code == 0
|
|
expected = " Band 1: [2, 4, 8] (method: 'nearest')\n Band 2: [2, 4, 8] (method: 'nearest')\n Band 3: [2, 4, 8] (method: 'nearest')\n"
|
|
assert result.output.endswith(expected)
|
|
|
|
|
|
def test_build_pow_ls(data):
|
|
runner = CliRunner()
|
|
inputfile = str(data.join('RGB.byte.tif'))
|
|
result = runner.invoke(cli, ['overview', inputfile, '--build', '2^1..3'])
|
|
assert result.exit_code == 0
|
|
result = runner.invoke(cli, ['overview', inputfile, '--ls'])
|
|
assert result.exit_code == 0
|
|
expected = " Band 1: [2, 4, 8] (method: 'nearest')\n Band 2: [2, 4, 8] (method: 'nearest')\n Band 3: [2, 4, 8] (method: 'nearest')\n"
|
|
assert result.output.endswith(expected)
|
|
|
|
|
|
def test_rebuild_ls(data):
|
|
runner = CliRunner()
|
|
inputfile = str(data.join('RGB.byte.tif'))
|
|
|
|
result = runner.invoke(
|
|
cli,
|
|
['overview', inputfile, '--build', '2,4,8', '--resampling', 'cubic'])
|
|
assert result.exit_code == 0
|
|
|
|
result = runner.invoke(cli, ['overview', inputfile, '--rebuild'])
|
|
assert result.exit_code == 0
|
|
|
|
result = runner.invoke(cli, ['overview', inputfile, '--ls'])
|
|
assert result.exit_code == 0
|
|
|
|
expected = " Band 1: [2, 4, 8] (method: 'cubic')\n Band 2: [2, 4, 8] (method: 'cubic')\n Band 3: [2, 4, 8] (method: 'cubic')\n"
|
|
assert result.output.endswith(expected)
|