mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
222 lines
6.9 KiB
Python
222 lines
6.9 KiB
Python
"""Tests of rasterio.path"""
|
|
|
|
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
import rasterio
|
|
from rasterio.errors import PathError
|
|
from rasterio.path import parse_path, vsi_path, ParsedPath, UnparsedPath
|
|
|
|
|
|
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("uri", ["/test.tif", "file:///test.tif"])
|
|
def test_parsed_path_not_remote(uri):
|
|
"""Check for paths that are not remote"""
|
|
assert not ParsedPath.from_uri(uri).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
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"uri", ["s3://bucket/test.tif", "https://example.com/test.tif"]
|
|
)
|
|
def test_parsed_path_not_local(uri):
|
|
"""Check for paths that are not local"""
|
|
assert not ParsedPath.from_uri(uri).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'
|
|
|
|
|
|
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'
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform == 'win32', reason="Checking behavior on posix, not win32")
|
|
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_path_as_vsi_scheme():
|
|
"""Correctly make a vsi path"""
|
|
assert ParsedPath('/foo.tif', 'tests/data/files.zip', 'zip').as_vsi() == '/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_path_as_vsi_unparsed():
|
|
"""Correctly make GDAL filename from unparsed path"""
|
|
assert UnparsedPath("foo").as_vsi() == "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/rasterio/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'
|
|
|
|
|
|
def test_vsi_path_zip():
|
|
"""A zip:// URLs vsi path is correct (see #1377)"""
|
|
url = 'zip:///path/to/zip/some.zip!path/to/file.txt'
|
|
assert vsi_path(parse_path(url)) == '/vsizip//path/to/zip/some.zip/path/to/file.txt'
|
|
|
|
|
|
def test_vsi_path_zip_plus_https():
|
|
"""A zip+https:// URLs vsi path is correct (see #1151)"""
|
|
url = 'zip+https://example.com/foo.zip!bar.tif'
|
|
assert vsi_path(parse_path(url)) == '/vsizip/vsicurl/https://example.com/foo.zip/bar.tif'
|
|
|
|
|
|
@pytest.mark.parametrize("path", ["DRIVER:/vsifoo/bar:var", "SENTINEL2_L1C:S2A_OPER_MTD_SAFL1C_PDMC_20150818T101440_R022_V20150813T102406_20150813T102406.xml:10m:EPSG_32632"])
|
|
def test_driver_prefixed_path(path):
|
|
parsed = parse_path(path)
|
|
assert isinstance(parsed, UnparsedPath)
|
|
|
|
|
|
@pytest.mark.parametrize("path", [0, -1.0, object()])
|
|
def test_path_error(path):
|
|
with pytest.raises(PathError):
|
|
parse_path(path)
|
|
|
|
|
|
def test_parse_path():
|
|
pathlib = pytest.importorskip("pathlib")
|
|
assert isinstance(parse_path(pathlib.Path("/foo/bar.tif")), ParsedPath)
|
|
|
|
|
|
def test_parse_path_win():
|
|
pathlib = pytest.importorskip("pathlib")
|
|
assert isinstance(parse_path(pathlib.PureWindowsPath(r"C:\foo\bar.tif")), ParsedPath)
|
|
|
|
|
|
def test_parse_path_win_no_pathlib(monkeypatch):
|
|
monkeypatch.setattr(rasterio.path.sys, "platform", "win32")
|
|
monkeypatch.setattr(rasterio.path, "pathlib", None)
|
|
assert isinstance(parse_path(r"C:\foo\bar.tif"), UnparsedPath)
|