Added tests

This commit is contained in:
Nelle Varoquaux
2011-12-22 12:27:06 +01:00
parent c6f16dbff5
commit 5d8110bab2
3 changed files with 28 additions and 3 deletions
+23
View File
@@ -0,0 +1,23 @@
import numpy as np
import unittest
from skimage.filter import harris_corner_detector
class TestHarris(unittest.TestCase):
def test_square_image(self):
im = np.zeros((50, 50)).astype(float)
im[:25, :25] = 1.
results = harris_corner_detector(im)
self.assertTrue(results.any())
self.assertTrue(len(results) == 1)
def test_noisy_square_image(self):
im = np.zeros((50, 50)).astype(float)
im[:25, :25] = 1.
im = im + np.random.uniform(size=im.shape) * .5
results = harris_corner_detector(im)
assert results.any()
assert len(results) == 1