1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
import pytest
import rasterio
from rasterio.profiles import default_gtiff_profile
def test_set_band_descriptions(tmpdir):
"""Descriptions 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.descriptions == (None, None, None)
dst.descriptions = ["this is a test band", "this is another test band", None]
assert dst.descriptions == (
"this is a test band", "this is another test band", None)
@pytest.mark.parametrize('value', [[], ['x'], ['x', 'y', 'z']])
def test_set_band_descriptions_error(tmpdir, value):
"""Number of descriptions 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.descriptions = value
|
