mirror of
https://github.com/rasterio/rasterio.git
synced 2026-01-25 14:10:17 +00:00
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
import pytest
|
|
|
|
import rasterio
|
|
from rasterio.profiles import default_gtiff_profile
|
|
|
|
|
|
def test_set_scales(tmpdir):
|
|
"""Scales can be set when dataset is open"""
|
|
tmptiff = str(tmpdir.join("test.tif"))
|
|
with rasterio.open(
|
|
tmptiff, "w", count=3, height=256, width=256, **default_gtiff_profile
|
|
) as dst:
|
|
assert dst.scales == (1.0,) * 3
|
|
dst.scales = [0.1] * 3
|
|
assert dst.scales == (0.1,) * 3
|
|
|
|
|
|
@pytest.mark.parametrize("value", [[0.1], [2.0] * 3, []])
|
|
def test_set_scales_error(tmpdir, value):
|
|
"""Number of values must match band count"""
|
|
tmptiff = str(tmpdir.join("test.tif"))
|
|
with rasterio.open(
|
|
tmptiff, "w", count=2, height=256, width=256, **default_gtiff_profile
|
|
) as dst:
|
|
with pytest.raises(ValueError):
|
|
dst.scales = value
|
|
|
|
|
|
def test_set_offsets(tmpdir):
|
|
"""Scales can be set when dataset is open"""
|
|
tmptiff = str(tmpdir.join("test.tif"))
|
|
with rasterio.open(
|
|
tmptiff, "w", count=3, height=256, width=256, **default_gtiff_profile
|
|
) as dst:
|
|
assert dst.offsets == (0.0,) * 3
|
|
dst.offsets = [0.1] * 3
|
|
assert dst.offsets == (0.1,) * 3
|
|
|
|
|
|
@pytest.mark.parametrize("value", [[0.1], [2.0] * 3, []])
|
|
def test_set_offsets_error(tmpdir, value):
|
|
"""Number of values must match band count"""
|
|
tmptiff = str(tmpdir.join("test.tif"))
|
|
with rasterio.open(
|
|
tmptiff, "w", count=2, height=256, width=256, **default_gtiff_profile
|
|
) as dst:
|
|
with pytest.raises(ValueError):
|
|
dst.offsets = value
|