wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-10 23:41:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add test cases for piecewise affine

Browse Source
This commit is contained in:
Johannes Schönberger
2012-09-03 22:05:12 +02:00
parent 677478873c
commit 05ae17e0d5
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/transform/tests/test_geometric.py
+10 -1
View File
@@ -3,7 +3,8 @@ from numpy.testing import assert_equal, assert_array_almost_equal
from skimage.transform._geometric import _stackcopy
from skimage.transform import (estimate_transform,
SimilarityTransform, AffineTransform,
ProjectiveTransform, PolynomialTransform)
ProjectiveTransform, PolynomialTransform,
PiecewiseAffineTransform)
SRC = np.array([
@@ -110,6 +111,14 @@ def test_affine_init():
assert_array_almost_equal(tform2.translation, translation)
def test_piecewise_affine():
tform = PiecewiseAffineTransform()
tform.estimate(SRC, DST)
# make sure each single affine transform is exactly estimated
assert_array_almost_equal(tform(SRC), DST)
assert_array_almost_equal(tform.inverse(DST), SRC)
def test_projective_estimation():
# exact solution
tform = estimate_transform('projective', SRC[:4, :], DST[:4, :])