mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
2308aa6d5e
The signatures of all cdef class initializers in rasterio._base and rasterio._io have changed: these method now take instance of ParsedPath or UnparsedPath from rasterio.path and do not take strings. New ParsedPath and UnparsedPath classes have been added to improve input file handling and validation. The existing parse_path and vsi_path functions have been rewritten to use these new classes and have been moved to the new rasterio.path module. The signature of rasterio.open has not been changed and users of Rasterio will be unaffected. The rasterio.vfs module and its functions are being replaced by new functions in rasterio.path.
159 lines
4.7 KiB
Python
159 lines
4.7 KiB
Python
import logging
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
import rasterio
|
|
from rasterio.profiles import default_gtiff_profile
|
|
from rasterio.path import parse_path, vsi_path, ParsedPath, UnparsedPath
|
|
|
|
|
|
logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)
|
|
|
|
|
|
def test_parsed_path_name():
|
|
"""A parsed path's name property is correct"""
|
|
assert ParsedPath('bar.tif', 'foo.zip', 'zip').name == 'zip://foo.zip!bar.tif'
|
|
|
|
|
|
def test_parsed_path_name_no_archive():
|
|
"""A parsed path's name property is correct"""
|
|
assert ParsedPath('bar.tif', None, 'file').name == 'file://bar.tif'
|
|
|
|
|
|
def test_parsed_path_name_no_scheme():
|
|
"""A parsed path's name property is correct"""
|
|
assert ParsedPath('bar.tif', None, None).name == 'bar.tif'
|
|
|
|
|
|
def test_unparsed_path_name():
|
|
"""An unparsed path's name property is correct"""
|
|
assert UnparsedPath('/vsifoo/bar/tif').name == '/vsifoo/bar/tif'
|
|
|
|
|
|
@pytest.mark.parametrize('scheme', ['s3', 'ftp', 'http', 'https', 'zip+s3'])
|
|
def test_parsed_path_remote(scheme):
|
|
"""A parsed path is remote"""
|
|
assert ParsedPath('example.com/foo.tif', None, scheme).is_remote
|
|
|
|
|
|
@pytest.mark.parametrize('scheme', [None, '', 'zip', 'tar', 'file', 'zip+file'])
|
|
def test_parsed_path_file_local(scheme):
|
|
"""A parsed path is remote"""
|
|
assert ParsedPath('foo.tif', None, scheme).is_local
|
|
|
|
|
|
def test_parse_path_zip():
|
|
"""Correctly parse zip scheme URL"""
|
|
parsed = parse_path('zip://tests/data/files.zip!foo.tif')
|
|
assert parsed.path == 'foo.tif'
|
|
assert parsed.archive == 'tests/data/files.zip'
|
|
assert parsed.scheme == 'zip'
|
|
|
|
|
|
def test_parse_path_zip_and_file():
|
|
"""Correctly parse zip+file scheme URL"""
|
|
parsed = parse_path('zip+file://tests/data/files.zip!foo.tif')
|
|
assert parsed.path == 'foo.tif'
|
|
assert parsed.archive == 'tests/data/files.zip'
|
|
assert parsed.scheme == 'zip+file'
|
|
|
|
|
|
def test_parse_path_file_scheme():
|
|
"""Correctly parse file:// URL"""
|
|
parsed = parse_path('file://foo.tif')
|
|
assert parsed.path == 'foo.tif'
|
|
assert parsed.archive is None
|
|
assert parsed.scheme == 'file'
|
|
|
|
|
|
def test_parse_path_file():
|
|
"""Correctly parse an ordinary filesystem path"""
|
|
parsed = parse_path('/foo.tif')
|
|
assert parsed.path == '/foo.tif'
|
|
assert parsed.archive is None
|
|
assert parsed.scheme is None
|
|
|
|
|
|
def test_parse_gdal_vsi():
|
|
"""GDAL dataset identifiers fall through properly"""
|
|
assert parse_path('/vsifoo/bar').path == '/vsifoo/bar'
|
|
|
|
|
|
def test_parse_gdal():
|
|
"""GDAL dataset identifiers fall through properly"""
|
|
assert parse_path('GDAL:filepath:varname').path == 'GDAL:filepath:varname'
|
|
|
|
|
|
def test_parse_windows_path(monkeypatch):
|
|
"""Return Windows paths unparsed"""
|
|
monkeypatch.setattr(sys, 'platform', 'win32')
|
|
assert parse_path(r'C:\\foo.tif').path == r'C:\\foo.tif'
|
|
|
|
|
|
def test_vsi_path_scheme():
|
|
"""Correctly make a vsi path"""
|
|
assert vsi_path(ParsedPath('/foo.tif', 'tests/data/files.zip', 'zip')) == '/vsizip/tests/data/files.zip/foo.tif'
|
|
|
|
|
|
def test_vsi_path_file():
|
|
"""Correctly make and ordinary file path from a file path"""
|
|
assert vsi_path(ParsedPath('foo.tif', None, 'file')) == 'foo.tif'
|
|
|
|
|
|
def test_vsi_path_curl():
|
|
"""Correctly make and ordinary file path from a https path"""
|
|
assert vsi_path(ParsedPath('example.com/foo.tif', None, 'https')) == '/vsicurl/https://example.com/foo.tif'
|
|
|
|
|
|
def test_vsi_path_unparsed():
|
|
"""Correctly make GDAL filename from unparsed path"""
|
|
assert vsi_path(UnparsedPath("foo")) == "foo"
|
|
|
|
|
|
def test_vsi_path_error():
|
|
"""Raise ValuError if argument is not a path"""
|
|
with pytest.raises(ValueError):
|
|
vsi_path("foo")
|
|
|
|
|
|
def test_read_vfs_zip():
|
|
with rasterio.open(
|
|
'zip://tests/data/files.zip!/RGB.byte.tif') as src:
|
|
assert src.name == 'zip://tests/data/files.zip!/RGB.byte.tif'
|
|
assert src.count == 3
|
|
|
|
|
|
def test_read_vfs_file():
|
|
with rasterio.open(
|
|
'file://tests/data/RGB.byte.tif') as src:
|
|
assert src.name == 'file://tests/data/RGB.byte.tif'
|
|
assert src.count == 3
|
|
|
|
def test_read_vfs_zip_cmp_array():
|
|
with rasterio.open(
|
|
'zip://tests/data/files.zip!/RGB.byte.tif') as src:
|
|
zip_arr = src.read()
|
|
|
|
with rasterio.open(
|
|
'file://tests/data/RGB.byte.tif') as src:
|
|
file_arr = src.read()
|
|
|
|
assert zip_arr.dumps() == file_arr.dumps()
|
|
|
|
|
|
def test_read_vfs_none():
|
|
with rasterio.open(
|
|
'tests/data/RGB.byte.tif') as src:
|
|
assert src.name == 'tests/data/RGB.byte.tif'
|
|
assert src.count == 3
|
|
|
|
|
|
def test_parse_path_accept_get_params():
|
|
# See https://github.com/mapbox/rasterio/issues/1121
|
|
parsed = parse_path('http://example.com/index?a=1')
|
|
assert isinstance(parsed, ParsedPath)
|
|
assert parsed.path == 'example.com/index?a=1'
|
|
assert parsed.archive is None
|
|
assert parsed.scheme == 'http'
|