Fix peak_local_max's output for ndarray when indices is set to False

This commit is contained in:
yangzetian
2013-04-01 20:34:47 +08:00
parent 11a2f1ebcc
commit f77def35d8
2 changed files with 8 additions and 1 deletions
+6
View File
@@ -116,6 +116,12 @@ def test_indices_with_labels():
indices=True, exclude_border=False)
assert (result == np.transpose(expected.nonzero())).all()
def test_ndarray_indices_false():
nd_image = np.zeros((5,5,5))
nd_image[2,2,2] = 1
peaks = peak.peak_local_max(nd_image, min_distance=1, indices=False)
assert (peaks == nd_image.astype(np.bool)).all()
if __name__ == '__main__':
from numpy import testing