mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-30 16:50:01 +08:00
Josh Warner (Mac)
105 lines
4.0 KiB
Python
105 lines
4.0 KiB
Python
import numpy as np
|
|
from numpy.testing import assert_array_equal, assert_allclose
|
|
|
|
from skimage.draw import ellipsoid, ellipsoid_stats
|
|
|
|
|
|
def test_ellipsoid_bool():
|
|
test = ellipsoid(2, 2, 2)[1:-1, 1:-1, 1:-1]
|
|
test_anisotropic = ellipsoid(2, 2, 4, spacing=(1., 1., 2.))
|
|
test_anisotropic = test_anisotropic[1:-1, 1:-1, 1:-1]
|
|
|
|
expected = np.array([[[0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0],
|
|
[0, 0, 1, 0, 0],
|
|
[0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0]],
|
|
|
|
[[0, 0, 0, 0, 0],
|
|
[0, 1, 1, 1, 0],
|
|
[0, 1, 1, 1, 0],
|
|
[0, 1, 1, 1, 0],
|
|
[0, 0, 0, 0, 0]],
|
|
|
|
[[0, 0, 1, 0, 0],
|
|
[0, 1, 1, 1, 0],
|
|
[1, 1, 1, 1, 1],
|
|
[0, 1, 1, 1, 0],
|
|
[0, 0, 1, 0, 0]],
|
|
|
|
[[0, 0, 0, 0, 0],
|
|
[0, 1, 1, 1, 0],
|
|
[0, 1, 1, 1, 0],
|
|
[0, 1, 1, 1, 0],
|
|
[0, 0, 0, 0, 0]],
|
|
|
|
[[0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0],
|
|
[0, 0, 1, 0, 0],
|
|
[0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0]]])
|
|
|
|
assert_array_equal(test, expected.astype(bool))
|
|
assert_array_equal(test_anisotropic, expected.astype(bool))
|
|
|
|
|
|
def test_ellipsoid_levelset():
|
|
test = ellipsoid(2, 2, 2, levelset=True)[1:-1, 1:-1, 1:-1]
|
|
test_anisotropic = ellipsoid(2, 2, 4, spacing=(1., 1., 2.),
|
|
levelset=True)
|
|
test_anisotropic = test_anisotropic[1:-1, 1:-1, 1:-1]
|
|
|
|
expected = np.array([[[ 2. , 1.25, 1. , 1.25, 2. ],
|
|
[ 1.25, 0.5 , 0.25, 0.5 , 1.25],
|
|
[ 1. , 0.25, 0. , 0.25, 1. ],
|
|
[ 1.25, 0.5 , 0.25, 0.5 , 1.25],
|
|
[ 2. , 1.25, 1. , 1.25, 2. ]],
|
|
|
|
[[ 1.25, 0.5 , 0.25, 0.5 , 1.25],
|
|
[ 0.5 , -0.25, -0.5 , -0.25, 0.5 ],
|
|
[ 0.25, -0.5 , -0.75, -0.5 , 0.25],
|
|
[ 0.5 , -0.25, -0.5 , -0.25, 0.5 ],
|
|
[ 1.25, 0.5 , 0.25, 0.5 , 1.25]],
|
|
|
|
[[ 1. , 0.25, 0. , 0.25, 1. ],
|
|
[ 0.25, -0.5 , -0.75, -0.5 , 0.25],
|
|
[ 0. , -0.75, -1. , -0.75, 0. ],
|
|
[ 0.25, -0.5 , -0.75, -0.5 , 0.25],
|
|
[ 1. , 0.25, 0. , 0.25, 1. ]],
|
|
|
|
[[ 1.25, 0.5 , 0.25, 0.5 , 1.25],
|
|
[ 0.5 , -0.25, -0.5 , -0.25, 0.5 ],
|
|
[ 0.25, -0.5 , -0.75, -0.5 , 0.25],
|
|
[ 0.5 , -0.25, -0.5 , -0.25, 0.5 ],
|
|
[ 1.25, 0.5 , 0.25, 0.5 , 1.25]],
|
|
|
|
[[ 2. , 1.25, 1. , 1.25, 2. ],
|
|
[ 1.25, 0.5 , 0.25, 0.5 , 1.25],
|
|
[ 1. , 0.25, 0. , 0.25, 1. ],
|
|
[ 1.25, 0.5 , 0.25, 0.5 , 1.25],
|
|
[ 2. , 1.25, 1. , 1.25, 2. ]]])
|
|
|
|
assert_allclose(test, expected)
|
|
assert_allclose(test_anisotropic, expected)
|
|
|
|
|
|
def test_ellipsoid_stats():
|
|
# Test comparison values generated by Wolfram Alpha
|
|
vol, surf = ellipsoid_stats(6, 10, 16)
|
|
assert(round(1280 * np.pi, 4) == round(vol, 4))
|
|
assert(1383.28 == round(surf, 2))
|
|
|
|
# Test when a <= b <= c does not hold
|
|
vol, surf = ellipsoid_stats(16, 6, 10)
|
|
assert(round(1280 * np.pi, 4) == round(vol, 4))
|
|
assert(1383.28 == round(surf, 2))
|
|
|
|
# Larger test to ensure reliability over broad range
|
|
vol, surf = ellipsoid_stats(17, 27, 169)
|
|
assert(round(103428 * np.pi, 4) == round(vol, 4))
|
|
assert(37426.3 == round(surf, 1))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
np.testing.run_module_suite()
|