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
|
"""Tests for rawpy plugin
"""
import imageio.v3 as iio
import pytest
import numpy as np
rawpy = pytest.importorskip("rawpy")
@pytest.mark.parametrize(
"im_in",
[
("Nikon_uncompressed.nef"),
("Blackmagic.dng"),
("Canon_Powershot.CRW"),
("Pentax_compressed.PEF"),
],
)
def test_read(test_images, im_in):
"""Test for reading .nef file from .test_images dir."""
# Construct image path
im_path = test_images / im_in
# Test if plugin's content mathces rawpy content
actual = iio.imread(im_path, index=..., plugin="rawpy")
expected = rawpy.imread(str(im_path)).postprocess()
assert np.allclose(actual, expected)
@pytest.mark.parametrize(
"im_in",
[
("Nikon_uncompressed.nef"),
("Blackmagic.dng"),
("Canon_Powershot.CRW"),
("Pentax_compressed.PEF"),
],
)
def test_read_with_default_index(test_images, im_in):
"""Test for reading .nef file from .test_images dir."""
# Construct image path
im_path = test_images / im_in
# Test if plugin's content mathces rawpy content
actual = iio.imread(im_path, plugin="rawpy")
expected = rawpy.imread(str(im_path)).postprocess()
assert np.allclose(actual, expected)
def test_iter(test_images):
"""Test for the iter function of rawpy plugin."""
# Construct image path
im_path = test_images / "Blackmagic.dng"
assert iio.imiter(im_path, plugin="rawpy")
def test_properties(test_images):
"""Test for reading properties of a raw image from .test_images dir."""
# Construct image path
im_path = test_images / "Nikon_uncompressed.nef"
# Test properties of a .nef image
properties = iio.improps(im_path, plugin="rawpy")
assert properties.shape == (593, 869)
assert properties.dtype == np.uint16
def test_metadata(test_images):
"""Test for reading metadata of a raw image from .test_images dir."""
# Construct image path
im_path = test_images / "Nikon_uncompressed.nef"
# Test metadata of a .nef image
metadata = iio.immeta(im_path, plugin="rawpy")
assert metadata["width"] == 869
assert metadata["height"] == 593
assert metadata["pixel_aspect"] == 1.0
|
