Reformat doctests so that tests pass

I believe the dtype print out was changed to a string in a recent version of numpy, so this change may actually break the doctest for older versions.
This commit is contained in:
Tony S Yu
2011-10-15 12:44:26 -04:00
parent 9e81cecbb6
commit 1cc0f36d15
+15 -15
View File
@@ -249,32 +249,32 @@ def is_local_maximum(image, labels=None, footprint=None):
>>> image[3, 3] = 1
>>> image
array([[ 0., 0., 0., 0.],
[ 0., 0., 2., 0.],
[ 0., 0., 0., 0.],
[ 0., 0., 0., 1.]])
[ 0., 0., 2., 0.],
[ 0., 0., 0., 0.],
[ 0., 0., 0., 1.]])
>>> is_local_maximum(image)
array([[ True, False, False, False],
[ True, False, True, False],
[ True, False, False, False],
[ True, True, False, True]], dtype=bool)
[ True, False, True, False],
[ True, False, False, False],
[ True, True, False, True]], dtype='bool')
>>> image = np.arange(16).reshape((4, 4))
>>> labels = np.array([[1, 2], [3, 4]])
>>> labels = np.repeat(np.repeat(labels, 2, axis=0), 2, axis=1)
>>> labels
array([[1, 1, 2, 2],
[1, 1, 2, 2],
[3, 3, 4, 4],
[3, 3, 4, 4]])
[1, 1, 2, 2],
[3, 3, 4, 4],
[3, 3, 4, 4]])
>>> image
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15]])
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15]])
>>> is_local_maximum(image, labels=labels)
array([[False, False, False, False],
[False, True, False, True],
[False, False, False, False],
[False, True, False, True]], dtype=bool)
[False, True, False, True],
[False, False, False, False],
[False, True, False, True]], dtype='bool')
"""
if labels is None:
labels = np.ones(image.shape, dtype=np.uint8)