-
Notifications
You must be signed in to change notification settings - Fork 523
/
test_blocks.py
221 lines (179 loc) · 8.08 KB
/
test_blocks.py
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
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
from functools import partial
import os.path
import shutil
import subprocess
import tempfile
import unittest
import numpy as np
import pytest
import rasterio
from rasterio import windows
from rasterio.errors import RasterBlockError
from rasterio.profiles import default_gtiff_profile
from .conftest import requires_gdal2
class WindowTest(unittest.TestCase):
def test_window_shape_None_start(self):
self.assertEqual(
rasterio.windows.shape(((None, 4), (None, 102))),
(4, 102))
def test_window_shape_None_stop(self):
self.assertEqual(
rasterio.windows.shape(((10, None), (10, None)), 100, 90),
(90, 80))
def test_window_shape_positive(self):
self.assertEqual(
rasterio.windows.shape(((0, 4), (1, 102))),
(4, 101))
def test_window_shape_negative(self):
self.assertEqual(
rasterio.windows.shape(((-10, None), (-10, None)), 100, 90),
(10, 10))
self.assertEqual(
rasterio.windows.shape(((~0, None), (~0, None)), 100, 90),
(1, 1))
self.assertEqual(
rasterio.windows.shape(((None, ~0), (None, ~0)), 100, 90),
(99, 89))
def test_eval(self):
self.assertEqual(
rasterio.windows.evaluate(((-10, None), (-10, None)), 100, 90),
windows.Window.from_slices((90, 100), (80, 90)))
self.assertEqual(
rasterio.windows.evaluate(((None, -10), (None, -10)), 100, 90),
windows.Window.from_slices((0, 90), (0, 80)))
def test_window_index():
idx = rasterio.windows.window_index(((0, 4), (1, 12)))
assert len(idx) == 2
r, c = idx
assert r.start == 0
assert r.stop == 4
assert c.start == 1
assert c.stop == 12
arr = np.ones((20, 20))
assert arr[idx].shape == (4, 11)
class RasterBlocksTest(unittest.TestCase):
def test_blocks(self):
with rasterio.open('tests/data/RGB.byte.tif') as s:
self.assertEqual(len(s.block_shapes), 3)
self.assertEqual(s.block_shapes, [(3, 791), (3, 791), (3, 791)])
itr = s.block_windows(1)
(j, i), first = next(itr)
self.assertEqual((j, i), (0, 0))
self.assertEqual(first, windows.Window.from_slices((0, 3), (0, 791)))
itr = s.block_windows()
(j, i), first = next(itr)
self.assertEqual((j, i), (0, 0))
self.assertEqual(first, windows.Window.from_slices((0, 3), (0, 791)))
(j, i), second = next(itr)
self.assertEqual((j, i), (1, 0))
self.assertEqual(second, windows.Window.from_slices((3, 6), (0, 791)))
(j, i), last = list(itr)[~0]
self.assertEqual((j, i), (239, 0))
self.assertEqual(
last, windows.Window.from_slices((717, 718), (0, 791)))
def test_block_coverage(self):
with rasterio.open('tests/data/RGB.byte.tif') as s:
self.assertEqual(
s.width * s.height,
sum(w.width * w.height for ji, w in s.block_windows(1)))
class WindowReadTest(unittest.TestCase):
def test_read_window(self):
with rasterio.open('tests/data/RGB.byte.tif') as s:
windows = s.block_windows(1)
ji, first_window = next(windows)
first_block = s.read(1, window=first_window)
self.assertEqual(first_block.dtype, rasterio.ubyte)
self.assertEqual(
first_block.shape,
rasterio.windows.shape(first_window))
class WindowWriteTest(unittest.TestCase):
def setUp(self):
self.tempdir = tempfile.mkdtemp()
def tearDown(self):
shutil.rmtree(self.tempdir)
@pytest.mark.gdalbin
def test_write_window(self):
name = os.path.join(self.tempdir, "test_write_window.tif")
a = np.ones((50, 50), dtype=rasterio.ubyte) * 127
with rasterio.open(
name, 'w',
driver='GTiff', width=100, height=100, count=1,
dtype=a.dtype) as s:
s.write(a, indexes=1, window=windows.Window(10, 30, 50, 50))
# subprocess.call(["open", name])
info = subprocess.check_output(["gdalinfo", "-stats", name])
self.assertTrue(
"Minimum=0.000, Maximum=127.000, "
"Mean=31.750, StdDev=54.993" in info.decode('utf-8'),
info)
def test_block_windows_unfiltered(path_rgb_byte_tif):
"""Get all block windows"""
with rasterio.open(path_rgb_byte_tif) as src:
assert len(list(src.block_windows())) == 240
def test_block_windows_filtered_all(path_rgb_byte_tif):
"""Get all block windows using filter"""
with rasterio.open(path_rgb_byte_tif) as src:
w, s, e, n = src.bounds
focus_window = src.window(w, s, e, n)
filter_func = partial(windows.intersect, focus_window)
itr = ((ij, win) for ij, win in src.block_windows() if filter_func(win))
assert len(list(itr)) == 240
def test_block_windows_filtered_one(path_rgb_byte_tif):
"""Get the first block windows using filter"""
with rasterio.open(path_rgb_byte_tif) as src:
w, s, e, n = src.bounds
focus_window = src.window(w, n - 1.0, w + 1.0, n)
filter_func = partial(windows.intersect, focus_window)
itr = ((ij, win) for ij, win in src.block_windows() if filter_func(win))
assert next(itr) == ((0, 0), windows.Window.from_slices((0, 3), (0, 791)))
with pytest.raises(StopIteration):
next(itr)
def test_block_windows_filtered_none(path_rgb_byte_tif):
"""Get no block windows using filter"""
with rasterio.open(path_rgb_byte_tif) as src:
w, s, e, n = src.bounds
focus_window = src.window(w - 100.0, n + 1.0, w - 1.0, n + 100.0)
filter_func = partial(windows.intersect, focus_window)
itr = ((ij, win) for ij, win in src.block_windows() if filter_func(win))
with pytest.raises(StopIteration):
next(itr)
@requires_gdal2(reason="TIFF block size access requires GDAL 2.0")
def test_block_size_tiff(path_rgb_byte_tif):
"""Without compression a TIFF's blocks are all the same size"""
with rasterio.open(path_rgb_byte_tif) as src:
block_windows = list(src.block_windows())
sizes = [src.block_size(1, i, j) for (i, j), w in block_windows]
assert sizes.count(2373) == 1
assert sizes.count(7119) == len(block_windows) - 1
@requires_gdal2(reason="TIFF block size access requires GDAL 2.0")
def test_block_size_exception():
"""A JPEG has no TIFF metadata and no API for block size"""
with pytest.raises(RasterBlockError):
with rasterio.open('tests/data/389225main_sw_1965_1024.jpg') as src:
src.block_size(1, 0, 0)
def test_block_window_tiff(path_rgb_byte_tif):
"""Block window accessors are consistent"""
with rasterio.open(path_rgb_byte_tif) as src:
for (i, j), w in src.block_windows():
assert src.block_window(1, i, j) == w
@pytest.mark.parametrize("blocksize", [16, 32, 256, 1024])
def test_block_windows_bigger_blocksize(tmpdir, blocksize):
"""Ensure that block sizes greater than raster size are ok"""
tempfile = str(tmpdir.join("test.tif"))
profile = default_gtiff_profile.copy()
profile.update(height=16, width=16, count=1, blockxsize=blocksize, blockysize=blocksize)
with rasterio.open(tempfile, "w", **profile) as dst:
assert dst.is_tiled
for ij, window in dst.block_windows():
dst.write(np.ones((1, 1), dtype="uint8"), 1, window=window)
with rasterio.open(tempfile) as dst:
assert list(dst.block_windows()) == [((0, 0), windows.Window(0, 0, 16, 16))]
assert (dst.read(1) == 1).all()
@pytest.mark.parametrize("blocksizes", [{"blockxsize": 33, "blockysize": 32}, {"blockxsize": 32, "blockysize": 33}])
def test_odd_blocksize_error(tmpdir, blocksizes):
"""For a tiled TIFF block sizes must be multiples of 16"""
tempfile = str(tmpdir.join("test.tif"))
profile = default_gtiff_profile.copy()
profile.update(height=64, width=64, count=1, **blocksizes)
with pytest.raises(RasterBlockError):
rasterio.open(tempfile, "w", **profile)