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
|
""" Tests for the shockwave flash plugin
"""
import numpy as np
import pytest
import imageio.v2 as iio
import imageio.plugins
from imageio import core
from imageio.core import IS_PYPY
from conftest import deprecated_test
def mean(x):
return x.sum() / x.size # pypy-compat mean
@deprecated_test
def test_format_selection(test_images, tmp_path):
fname1 = test_images / "stent.swf"
fname2 = tmp_path / "stent.out.swf"
F = iio.formats["swf"]
assert F.name == "SWF"
assert type(iio.formats[".swf"]) is type(F)
assert type(iio.read(fname1).format) is type(F)
assert type(iio.save(fname2).format) is type(F)
def test_reading_saving(test_images, tmp_path):
fname1 = test_images / "stent.swf"
fname2 = tmp_path / "stent.out.swf"
fname3 = tmp_path / "stent.compressed.swf"
fname4 = tmp_path / "stent.out2.swf"
# Read
R = iio.read(fname1)
assert len(R) == 10
assert R.get_meta_data() == {} # always empty dict
ims1 = []
for im in R:
assert im.shape == (657, 451, 4)
assert mean(im) > 0
ims1.append(im)
# Seek
assert (R.get_data(3) == ims1[3]).all()
# Fails
with pytest.raises(IndexError):
R.get_data(-1) # No negative index
with pytest.raises(IndexError):
R.get_data(10) # Out of bounds
R.close()
# Test loop
R = iio.read(fname1, loop=True)
assert (R.get_data(10) == ims1[0]).all()
# setting meta data is ignored
W = iio.save(fname2)
W.set_meta_data({"foo": 3})
W.close()
# Just make sure mimread works
assert len(iio.mimread(fname1)) == 10
# I'm not sure why, but the below does not work on pypy, which is weird,
# because the file *is* closed, but somehow it's not flushed? Ah well ...
if IS_PYPY:
return
# Write and re-read, now without loop, and with html page
iio.mimsave(fname2, ims1, loop=False, html=True)
ims2 = iio.mimread(fname2)
# Check images. We can expect exact match, since
# SWF is lossless.
assert len(ims1) == len(ims2)
for im1, im2 in zip(ims1, ims2):
assert (im1 == im2).all()
# Test compressed
iio.mimsave(fname3, ims2, compress=True)
ims3 = iio.mimread(fname3)
assert len(ims1) == len(ims3)
for im1, im3 in zip(ims1, ims3):
assert (im1 == im3).all()
# Test conventional, Bonus, we don't officially support this.
_swf = imageio.plugins.swf.load_lib()
_swf.write_swf(fname4, ims1)
ims4 = _swf.read_swf(fname4)
assert len(ims1) == len(ims4)
for im1, im4 in zip(ims1, ims4):
assert (im1 == im4).all()
# We want to manually validate that this file plays in 3d party tools
# So we write a small HTML5 doc that we can load
html = """<!DOCTYPE html>
<html>
<body>
Original:
<embed src="%s">
<br ><br >
Written:
<embed src="%s">
<br ><br >
Compressed:
<embed src="%s">
<br ><br >
Written 2:
<embed src="%s">
</body>
</html>
""" % (
fname1,
fname2,
fname3,
fname4,
)
with open(tmp_path / "test_swf.html", "wb") as f:
for line in html.splitlines():
f.write(line.strip().encode("utf-8") + b"\n")
@pytest.mark.needs_internet
def test_read_from_url():
burl = "https://raw.githubusercontent.com/imageio/imageio-binaries/master/"
url = burl + "images/stent.swf"
ims = iio.mimread(url)
assert len(ims) == 10
@deprecated_test
def test_invalid(test_images, tmp_path):
fname2 = tmp_path / "stent.invalid.swf"
# Empty file
with open(fname2, "wb"):
pass
assert not iio.formats.search_read_format(core.Request(fname2, "rI"))
with pytest.raises(RuntimeError):
iio.mimread(fname2, "swf")
# File with BS data
with open(fname2, "wb") as f:
f.write(b"x" * 100)
assert not iio.formats.search_read_format(core.Request(fname2, "rI"))
with pytest.raises(RuntimeError):
iio.mimread(fname2, "swf")
@pytest.mark.needs_internet
def test_lowlevel():
# Some tests from low level implementation that is not covered
# by using the plugin itself.
_swf = imageio.plugins.swf.load_lib()
tag = _swf.Tag()
with pytest.raises(NotImplementedError):
tag.process_tag()
assert tag.make_matrix_record() == "00000000"
assert tag.make_matrix_record(scale_xy=(1, 1))
assert tag.make_matrix_record(rot_xy=(1, 1))
assert tag.make_matrix_record(trans_xy=(1, 1))
SetBackgroundTag = _swf.SetBackgroundTag
assert SetBackgroundTag(1, 2, 3).rgb == SetBackgroundTag((1, 2, 3)).rgb
tag = _swf.ShapeTag(0, (0, 0), (1, 1))
assert tag.make_style_change_record(1, 1, (1, 1))
assert tag.make_style_change_record()
assert (
tag.make_straight_edge_record(2, 3).tobytes()
== tag.make_straight_edge_record((2, 3)).tobytes()
)
def test_types(test_images, tmp_path):
fname2 = tmp_path / "stent.out3.swf"
for dtype in [
np.uint8,
np.uint16,
np.uint32,
np.uint64,
np.int8,
np.int16,
np.int32,
np.int64,
np.float16,
np.float32,
np.float64,
]:
for shape in [(100, 1), (100, 3)]:
# Repeats an identity matrix, just for testing
im1 = np.dstack((np.identity(shape[0], dtype=dtype),) * shape[1])
iio.mimsave(fname2, [im1], "swf")
im2 = iio.mimread(fname2, "swf")[0]
assert im2.shape == (100, 100, 4)
assert im2.dtype == np.uint8
if len(shape) == 3 and dtype == np.uint8:
assert (im1[:, :, 0] == im2[:, :, 0]).all()
|
