wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-12 06:37:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

TST: Add unit test for mesh orientation correction.

Browse Source
This commit is contained in:
Josh Warner (Mac)
2014-03-13 23:34:46 -05:00
parent f2b13047f7
commit e9ac84b77c
1 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/measure/tests/test_marching_cubes.py
+22 -1
View File
@@ -2,7 +2,8 @@ import numpy as np
from numpy.testing import assert_raises
from skimage.draw import ellipsoid, ellipsoid_stats
from skimage.measure import marching_cubes, mesh_surface_area
from skimage.measure import (marching_cubes, mesh_surface_area,
correct_mesh_orientation)
def test_marching_cubes_isotropic():
@@ -36,5 +37,25 @@ def test_invalid_input():
assert_raises(ValueError, marching_cubes, np.zeros((20, 20)), 0)
def test_correct_mesh_orientation():
sphere_small = ellipsoid(1, 1, 1, levelset=True)
verts, faces = marching_cubes(sphere_small, 0.)
# Correct mesh orientation - descent
corrected_faces1 = correct_mesh_orientation(sphere_small, verts, faces,
gradient_direction='descent')
corrected_faces2 = correct_mesh_orientation(sphere_small, verts, faces,
gradient_direction='ascent')
# Ensure ascent is opposite of descent for all faces
np.testing.assert_array_equal(corrected_faces1, corrected_faces2[:, ::-1])
# Ensure correct faces have been reversed: 1, 4, and 5
idx = [1, 4, 5]
expected = faces.copy()
expected[idx] = expected[idx, ::-1]
np.testing.assert_array_equal(expected, corrected_faces1)
if __name__ == '__main__':
np.testing.run_module_suite()