Add test case for corner_peaks function

This commit is contained in:
Johannes Schönberger
2012-12-09 18:26:09 +01:00
parent a7411bc678
commit 377ae3b968
+12 -1
View File
@@ -5,7 +5,7 @@ from skimage import data
from skimage import img_as_float
from skimage.feature import (corner_moravec, corner_harris, corner_shi_tomasi,
corner_subpix, peak_local_max)
corner_subpix, peak_local_max, corner_peaks)
def test_square_image():
@@ -99,6 +99,17 @@ def test_subpix():
assert_array_equal(subpix[0], (24.5, 24.5))
def test_corner_peaks():
response = np.zeros((5, 5))
response[2:4, 2:4] = 1
corners = corner_peaks(response, exclude_border=False)
assert len(corners) == 1
corners = corner_peaks(response, exclude_border=False, min_distance=0)
assert len(corners) == 4
if __name__ == '__main__':
from numpy import testing
testing.run_module_suite()