rasterio/tests/test_open_options.py
2017-11-29 08:59:31 -08:00

45 lines
1.4 KiB
Python

"""Tests of dataset opening options and driver choice"""
import pytest
import rasterio
from rasterio.transform import Affine
from .conftest import requires_only_gdal1, requires_gdal2, requires_gdal22
@requires_only_gdal1
def test_driver_option_not_implemented():
"""Driver selection is not supported by GDAL 1.x"""
with pytest.raises(rasterio.errors.GDALOptionNotImplementedError):
rasterio.open('tests/data/RGB.byte.tif', driver='BMP')
@requires_only_gdal1
def test_open_options_not_implemented():
"""Open options are not supported by GDAL 1.x"""
with pytest.raises(rasterio.errors.GDALOptionNotImplementedError):
rasterio.open('tests/data/RGB.byte.tif', NUM_THREADS=42)
@requires_gdal2
def test_fail_with_missing_driver():
"""Fail to open a GeoTIFF without the GTiff driver"""
with pytest.raises(rasterio.errors.RasterioIOError):
rasterio.open('tests/data/RGB.byte.tif', driver='BMP')
@requires_gdal2
def test_open_specific_driver():
"""Open a GeoTIFF with the GTiff driver"""
with rasterio.open('tests/data/RGB.byte.tif', driver='GTiff') as src:
assert src.count == 3
@requires_gdal22
def test_open_specific_driver_with_options():
"""Open a GeoTIFF with the GTiff driver and GEOREF_SOURCES option"""
with rasterio.open(
'tests/data/RGB.byte.tif', driver='GTiff', GEOREF_SOURCES='NONE') as src:
assert src.transform == Affine.identity()