From 69dccda7cc87854e60c311edae4d7ba8d15b2da4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Johannes=20Sch=C3=B6nberger?= Date: Fri, 29 Nov 2013 23:03:57 +0100 Subject: [PATCH] Add test case for border keypoints --- skimage/feature/tests/test_brief.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/skimage/feature/tests/test_brief.py b/skimage/feature/tests/test_brief.py index 61042119..0f0a9b1f 100644 --- a/skimage/feature/tests/test_brief.py +++ b/skimage/feature/tests/test_brief.py @@ -7,14 +7,14 @@ from skimage.feature import (BRIEF, match_binary_descriptors, corner_peaks, corner_harris) -def test_descriptor_brief_color_image_unsupported_error(): +def test_color_image_unsupported_error(): """Brief descriptors can be evaluated on gray-scale images only.""" img = np.zeros((20, 20, 3)) keypoints = np.asarray([[7, 5], [11, 13]]) assert_raises(ValueError, BRIEF().extract, img, keypoints) -def test_descriptor_brief_normal_mode(): +def test_normal_mode(): """Verify the computed BRIEF descriptors with expected for normal mode.""" img = rgb2gray(data.lena()) @@ -36,7 +36,7 @@ def test_descriptor_brief_normal_mode(): assert_array_equal(descriptors, expected) -def test_descriptor_brief_uniform_mode(): +def test_uniform_mode(): """Verify the computed BRIEF descriptors with expected for uniform mode.""" img = rgb2gray(data.lena()) @@ -58,6 +58,17 @@ def test_descriptor_brief_uniform_mode(): assert_array_equal(descriptors, expected) +def test_border(): + img = np.zeros((100, 100)) + keypoints = np.array([[1, 1], [20, 20], [50, 50], [80, 80]]) + + extractor = BRIEF(patch_size=41) + descs, mask = extractor.extract(img, keypoints) + + assert descs.shape[0] == 3 + assert_array_equal(mask, (False, True, True, True)) + + if __name__ == '__main__': from numpy import testing testing.run_module_suite()