mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
"""High level tests for Rasterio's ``GDALDataset`` abstractions."""
|
|
|
|
|
|
from pathlib import Path
|
|
try:
|
|
from unittest.mock import MagicMock
|
|
except ImportError:
|
|
from mock import MagicMock
|
|
|
|
from numpy.testing import assert_almost_equal
|
|
import pytest
|
|
|
|
import rasterio
|
|
from rasterio.enums import Compression
|
|
from rasterio.errors import RasterioIOError, DatasetAttributeError
|
|
from rasterio.transform import Affine
|
|
|
|
|
|
def test_files(data):
|
|
tif = Path(data).joinpath('RGB.byte.tif')
|
|
aux = tif.parent.joinpath(tif.name + '.aux.xml')
|
|
with open(aux, 'w'):
|
|
pass
|
|
with rasterio.open(tif) as src:
|
|
assert src.files == [tif.as_posix(), aux.as_posix()]
|
|
|
|
|
|
def test_handle_closed(path_rgb_byte_tif):
|
|
"""Code that calls ``DatasetBase.handle()`` after it has been closed
|
|
should raise an exception.
|
|
"""
|
|
with rasterio.open(path_rgb_byte_tif) as src:
|
|
pass
|
|
with pytest.raises(RasterioIOError):
|
|
src.files
|
|
|
|
|
|
@pytest.mark.parametrize('tag_value', [item.value for item in Compression])
|
|
def test_dataset_compression(path_rgb_byte_tif, tag_value):
|
|
"""Compression is found from tags"""
|
|
with rasterio.open(path_rgb_byte_tif) as dataset:
|
|
dataset.tags = MagicMock()
|
|
dataset.tags.return_value = {'COMPRESSION': tag_value}
|
|
assert dataset.compression == Compression(tag_value)
|
|
|
|
|
|
def test_untiled_dataset_blocksize(tmpdir):
|
|
"""Blocksize is not relevant to untiled datasets (see #1689)"""
|
|
tmpfile = str(tmpdir.join("test.tif"))
|
|
with rasterio.open(
|
|
tmpfile, "w", driver="GTiff", count=1, height=13, width=23, dtype="uint8", crs="epsg:3857",
|
|
transform=Affine.identity(), blockxsize=64, blockysize=64) as dataset:
|
|
pass
|
|
|
|
with rasterio.open(tmpfile) as dataset:
|
|
assert not dataset.profile["tiled"]
|
|
assert dataset.shape == (13, 23)
|
|
assert dataset.block_shapes == [(13, 23)]
|
|
|
|
|
|
def test_dataset_readonly_attributes(path_rgb_byte_tif):
|
|
"""Attempts to set read-only attributes fail with DatasetAttributeError"""
|
|
with pytest.raises(DatasetAttributeError):
|
|
with rasterio.open(path_rgb_byte_tif) as dataset:
|
|
dataset.crs = "foo"
|
|
|
|
|
|
def test_dataset_readonly_attributes(path_rgb_byte_tif):
|
|
"""Attempts to set read-only attributes still fail with NotImplementedError"""
|
|
with pytest.raises(NotImplementedError):
|
|
with rasterio.open(path_rgb_byte_tif) as dataset:
|
|
dataset.crs = "foo"
|
|
|
|
|
|
def test_statistics(path_rgb_byte_tif):
|
|
"""Compute, store, and return basic statistics."""
|
|
with rasterio.open(path_rgb_byte_tif) as dataset:
|
|
stats = dataset.statistics(1)
|
|
assert stats.min == 0
|
|
assert stats.max == 255
|
|
assert_almost_equal(stats.mean, 29.947726688477)
|
|
assert_almost_equal(stats.std, 52.340921626611)
|
|
|
|
|
|
@pytest.mark.parametrize("blockysize", [1, 2, 3, 7, 61, 62])
|
|
def test_creation_untiled_blockysize(tmp_path, blockysize):
|
|
"""Check for fix of gh-2599"""
|
|
tmpfile = tmp_path / "test.tif"
|
|
with rasterio.open(
|
|
tmpfile,
|
|
"w",
|
|
count=1,
|
|
height=61,
|
|
width=37,
|
|
dtype="uint8",
|
|
blockysize=blockysize,
|
|
tiled=False,
|
|
) as dataset:
|
|
pass
|
|
|
|
with rasterio.open(tmpfile) as dataset:
|
|
assert not dataset.is_tiled
|
|
assert dataset.profile["blockysize"] == min(blockysize, 61)
|
|
assert dataset.block_shapes[0][0] == min(blockysize, 61)
|