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
|
# Copyright (c) DataLab Platform Developers, BSD 3-Clause license, see LICENSE file.
"""
Unit tests for edge detection computation functions.
"""
from __future__ import annotations
from typing import Callable
import pytest
from skimage import feature, filters, util
import sigima.objects
import sigima.params
import sigima.proc.image
from sigima.tests.data import get_test_image
from sigima.tests.helpers import check_array_result
@pytest.mark.validation
def test_canny() -> None:
"""Validation test for the image Canny edge detection processing."""
# See [1] in sigima\tests\image\__init__.py for more details about the validation.
src = get_test_image("flower.npy")
p = sigima.params.CannyParam.create(
sigma=1.0, low_threshold=0.1, high_threshold=0.2
)
dst = sigima.proc.image.canny(src, p)
exp = util.img_as_ubyte(
feature.canny(
src.data,
sigma=p.sigma,
low_threshold=p.low_threshold,
high_threshold=p.high_threshold,
use_quantiles=p.use_quantiles,
mode=p.mode,
cval=p.cval,
)
)
check_array_result(
f"Canny[sigma={p.sigma},low_threshold={p.low_threshold},"
f"high_threshold={p.high_threshold}]",
dst.data,
exp,
)
def __generic_edge_validation(func: Callable[[], sigima.objects.ImageObj]) -> None:
"""Generic test for edge detection methods."""
# See [1] in sigima\tests\image\__init__.py for more details about the validation.
src = get_test_image("flower.npy")
dst: sigima.objects.ImageObj = func(src)
exp = getattr(filters, func.__name__)(src.data)
check_array_result(f"{func.__name__.capitalize()}", dst.data, exp)
@pytest.mark.validation
def test_roberts() -> None:
"""Validation test for the image Roberts edge detection processing."""
__generic_edge_validation(sigima.proc.image.roberts)
@pytest.mark.validation
def test_prewitt() -> None:
"""Validation test for the image Prewitt edge detection processing."""
__generic_edge_validation(sigima.proc.image.prewitt)
@pytest.mark.validation
def test_prewitt_h() -> None:
"""Validation test for the image horizontal Prewitt edge detection processing."""
__generic_edge_validation(sigima.proc.image.prewitt_h)
@pytest.mark.validation
def test_prewitt_v() -> None:
"""Validation test for the image vertical Prewitt edge detection processing."""
__generic_edge_validation(sigima.proc.image.prewitt_v)
@pytest.mark.validation
def test_sobel() -> None:
"""Validation test for the image Sobel edge detection processing."""
__generic_edge_validation(sigima.proc.image.sobel)
@pytest.mark.validation
def test_sobel_h() -> None:
"""Validation test for the image horizontal Sobel edge detection processing."""
__generic_edge_validation(sigima.proc.image.sobel_h)
@pytest.mark.validation
def test_sobel_v() -> None:
"""Validation test for the image vertical Sobel edge detection processing."""
__generic_edge_validation(sigima.proc.image.sobel_v)
@pytest.mark.validation
def test_scharr() -> None:
"""Validation test for the image Scharr edge detection processing."""
__generic_edge_validation(sigima.proc.image.scharr)
@pytest.mark.validation
def test_scharr_h() -> None:
"""Validation test for the image horizontal Scharr edge detection processing."""
__generic_edge_validation(sigima.proc.image.scharr_h)
@pytest.mark.validation
def test_scharr_v() -> None:
"""Validation test for the image vertical Scharr edge detection processing."""
__generic_edge_validation(sigima.proc.image.scharr_v)
@pytest.mark.validation
def test_farid() -> None:
"""Validation test for the image Farid edge detection processing."""
__generic_edge_validation(sigima.proc.image.farid)
@pytest.mark.validation
def test_farid_h() -> None:
"""Validation test for the image horizontal Farid edge detection processing."""
__generic_edge_validation(sigima.proc.image.farid_h)
@pytest.mark.validation
def test_farid_v() -> None:
"""Validation test for the image vertical Farid edge detection processing."""
__generic_edge_validation(sigima.proc.image.farid_v)
@pytest.mark.validation
def test_laplace() -> None:
"""Validation test for the image Laplace edge detection processing."""
__generic_edge_validation(sigima.proc.image.laplace)
if __name__ == "__main__":
test_canny()
test_roberts()
test_prewitt()
test_prewitt_h()
test_prewitt_v()
test_sobel()
test_sobel_h()
test_sobel_v()
test_scharr()
test_scharr_h()
test_scharr_v()
test_farid()
test_farid_h()
test_farid_v()
test_laplace()
|
