mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-02 12:14:59 +08:00
add test case for subdivision of polygons
This commit is contained in:
committed by
Stefan van der Walt
parent
76c3574755
commit
f267f65666
@@ -1,15 +1,16 @@
|
||||
import numpy as np
|
||||
from skimage.measure import approximate_polygon
|
||||
from skimage.measure import approximate_polygon, subdivide_polygon
|
||||
|
||||
|
||||
square = np.array([
|
||||
[0, 0], [0, 1], [0, 2], [0, 3],
|
||||
[1, 3], [2, 3], [3, 3],
|
||||
[3, 2], [3, 1], [3, 0],
|
||||
[2, 0], [1, 0], [0, 0]
|
||||
])
|
||||
|
||||
|
||||
def test_approximate_polygon():
|
||||
square = np.array([
|
||||
[0, 0], [0, 1], [0, 2], [0, 3],
|
||||
[1, 3], [2, 3], [3, 3],
|
||||
[3, 2], [3, 1], [3, 0],
|
||||
[2, 0], [1, 0], [0, 0]
|
||||
])
|
||||
|
||||
out = approximate_polygon(square, 0.1)
|
||||
np.testing.assert_array_equal(out, square[(0, 3, 6, 9, 12), :])
|
||||
|
||||
@@ -23,5 +24,12 @@ def test_approximate_polygon():
|
||||
np.testing.assert_array_equal(out, square)
|
||||
|
||||
|
||||
def test_subdivide_polygon():
|
||||
for degree in range(1, 7):
|
||||
out = subdivide_polygon(square, degree)
|
||||
np.testing.assert_array_equal(out[-1], out[0])
|
||||
np.testing.assert_equal(out.shape[0], 2 * square.shape[0] - 1)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
np.testing.run_module_suite()
|
||||
|
||||
Reference in New Issue
Block a user