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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
"""Tests of _env util module"""
import pytest
from rasterio._env import (
GDALDataFinder,
PROJDataFinder,
get_proj_data_search_paths,
get_gdal_data,
)
from .conftest import gdal_version
@pytest.fixture
def mock_wheel(tmpdir):
"""A fake rasterio wheel"""
moduledir = tmpdir.mkdir("rasterio")
moduledir.ensure("__init__.py")
moduledir.ensure("_env.py")
moduledir.ensure("gdal_data/gdalvrt.xsd")
moduledir.ensure("proj_data/epsg")
return moduledir
@pytest.fixture
def mock_fhs(tmpdir):
"""A fake FHS system"""
tmpdir.ensure("share/gdal/gdalvrt.xsd")
tmpdir.ensure("share/proj/epsg")
return tmpdir
@pytest.fixture
def mock_debian(tmpdir):
"""A fake Debian multi-install system"""
tmpdir.ensure("share/gdal/3.8/gdalvrt.xsd")
tmpdir.ensure("share/gdal/3.9/gdalvrt.xsd")
tmpdir.ensure("share/gdal/3.10/gdalvrt.xsd")
tmpdir.ensure("share/gdal/3.11/gdalvrt.xsd")
tmpdir.ensure("share/gdal/3.12/gdalvrt.xsd")
tmpdir.ensure(f"share/gdal/{gdal_version.major}.{gdal_version.minor}/gdalvrt.xsd")
tmpdir.ensure("share/proj/epsg")
return tmpdir
def test_search_wheel_gdal_data_failure(tmpdir):
"""Fail to find GDAL data in a non-wheel"""
finder = GDALDataFinder()
assert not finder.search_wheel(str(tmpdir))
def test_search_wheel_gdal_data(mock_wheel):
"""Find GDAL data in a wheel"""
finder = GDALDataFinder()
assert finder.search_wheel(str(mock_wheel.join("_env.py"))) == str(mock_wheel.join("gdal_data"))
def test_search_prefix_gdal_data_failure(tmpdir):
"""Fail to find GDAL data in a bogus prefix"""
finder = GDALDataFinder()
assert not finder.search_prefix(str(tmpdir))
def test_search_prefix_gdal_data(mock_fhs):
"""Find GDAL data under prefix"""
finder = GDALDataFinder()
assert finder.search_prefix(str(mock_fhs)) == str(mock_fhs.join("share").join("gdal"))
def test_search_debian_gdal_data_failure(tmpdir):
"""Fail to find GDAL data in a bogus Debian location"""
finder = GDALDataFinder()
assert not finder.search_debian(str(tmpdir))
def test_search_debian_gdal_data(mock_debian):
"""Find GDAL data under Debian locations"""
finder = GDALDataFinder()
assert finder.search_debian(str(mock_debian)) == str(
mock_debian.join("share").join("gdal").join(f"{gdal_version.major}.{gdal_version.minor}")
)
def test_search_gdal_data_wheel(mock_wheel):
finder = GDALDataFinder()
assert finder.search(str(mock_wheel.join("_env.py"))) == str(mock_wheel.join("gdal_data"))
def test_search_gdal_data_fhs(mock_fhs):
finder = GDALDataFinder()
assert finder.search(str(mock_fhs)) == str(mock_fhs.join("share").join("gdal"))
def test_search_gdal_data_debian(mock_debian):
"""Find GDAL data under Debian locations"""
finder = GDALDataFinder()
assert finder.search(str(mock_debian)) == str(
mock_debian.join("share").join("gdal").join(f"{gdal_version.major}.{gdal_version.minor}")
)
def test_search_wheel_proj_data_failure(tmpdir):
"""Fail to find GDAL data in a non-wheel"""
finder = PROJDataFinder()
assert not finder.search_wheel(str(tmpdir))
def test_search_wheel_proj_data(mock_wheel):
"""Find GDAL data in a wheel"""
finder = PROJDataFinder()
assert finder.search_wheel(str(mock_wheel.join("_env.py"))) == str(mock_wheel.join("proj_data"))
def test_search_prefix_proj_data_failure(tmpdir):
"""Fail to find GDAL data in a bogus prefix"""
finder = PROJDataFinder()
assert not finder.search_prefix(str(tmpdir))
def test_search_prefix_proj_data(mock_fhs):
"""Find GDAL data under prefix"""
finder = PROJDataFinder()
assert finder.search_prefix(str(mock_fhs)) == str(mock_fhs.join("share").join("proj"))
def test_search_proj_data_wheel(mock_wheel):
finder = PROJDataFinder()
assert finder.search(str(mock_wheel.join("_env.py"))) == str(mock_wheel.join("proj_data"))
def test_search_proj_data_fhs(mock_fhs):
finder = PROJDataFinder()
assert finder.search(str(mock_fhs)) == str(mock_fhs.join("share").join("proj"))
def test_get_proj_data_search_paths():
assert isinstance(get_proj_data_search_paths(), list)
def test_get_gdal_data():
gdal_data = get_gdal_data()
if gdal_data is not None:
assert isinstance(gdal_data, str) and gdal_data
|
