updated test cases for adaptive thresholding

This commit is contained in:
Johannes Schönberger
2012-04-25 18:25:19 +02:00
parent f8e3ba42ef
commit bacde61e39
+12 -1
View File
@@ -37,6 +37,17 @@ class TestSimpleImage():
assert_array_equal(ref, out)
def test_threshold_adaptive_mean(self):
ref = np.array(
[[False, False, False, False, True],
[False, False, True, False, True],
[False, False, True, True, False],
[False, True, True, False, False],
[ True, True, False, False, False]]
)
out = threshold_adaptive(self.image, 3, 0, 'mean')
assert_array_equal(ref, out)
def test_threshold_adaptive_median(self):
ref = np.array(
[[False, False, False, False, True],
[False, False, True, False, False],
@@ -44,7 +55,7 @@ class TestSimpleImage():
[False, False, True, True, False],
[False, True, False, False, False]]
)
out = threshold_adaptive(self.image, 3, 0, 'mean')
out = threshold_adaptive(self.image, 3, 0, 'median')
assert_array_equal(ref, out)