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
|
import pytest
import imageio.v2 as iio
import numpy as np
def test_reader(test_images):
with iio.get_reader(test_images / "chelsea.png") as file:
assert file.request.extension == ".png"
img = file.get_data(0)
assert img.shape == (300, 451, 3)
# read using alternative method
file.set_image_index(0)
img2 = file.get_next_data()
assert np.allclose(img2, img)
def test_get_reader_format(test_images):
with iio.get_reader(test_images / "chelsea.png") as file:
with pytest.raises(TypeError):
file.format # v3 Pillow Plugin
def test_get_length(test_images):
with iio.get_reader(test_images / "chelsea.png") as file:
v3_count = file.instance.properties(index=...).shape[0]
assert len(file) == v3_count
def test_iterating(test_images):
with iio.get_reader(test_images / "newtonscradle.gif") as file:
for count, img in enumerate(file):
assert img.shape == (150, 200, 3)
assert count == len(file) - 1
def test_writer(test_images, tmp_path):
img = iio.imread(test_images / "chelsea.png")
with iio.get_writer(tmp_path / "test.png") as file:
assert file.request.extension == ".png"
file.append_data(img)
def test_get_writer_format(tmp_path):
with iio.get_writer(tmp_path / "test.png") as file:
with pytest.raises(TypeError):
file.format # v3 Pillow Plugin
def test_warnings(tmp_path, test_images):
img = iio.imread(test_images / "chelsea.png")
with iio.get_writer(tmp_path / "test.png") as writer:
with pytest.warns(UserWarning):
writer.append_data(img, meta={"meta_key": "meta_value"})
def test_mimwrite_exception(tmp_path):
img = np.zeros((4, 4), dtype=np.uint8)
with pytest.raises(ValueError):
iio.mimwrite(tmp_path / "test.png", img)
def test_kwarg_forwarding(tmp_path):
rng = np.random.default_rng()
# Write the frames to an animated GIF
with iio.get_writer(
tmp_path / "tmp.gif", duration=1000, loop=0, pilmode="RGB"
) as writer:
for _ in range(10):
frame = rng.integers(0, 255, (100, 100, 3), dtype=np.uint8)
writer.append_data(frame)
imgs = np.stack(iio.mimread(tmp_path / "tmp.gif"))
assert imgs.shape == (10, 100, 100, 3)
with iio.get_reader(tmp_path / "tmp.gif", pilmode="L") as file:
img = file.get_next_data()
assert img.shape == (100, 100)
assert img.meta["duration"] == 1000
assert img.meta["loop"] == 0
|
