wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-14 11:18:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

extend test cases for num_peaks parameter of peak_local_max

Browse Source
This commit is contained in:
Johannes Schönberger
2012-08-14 07:59:54 +02:00
parent a87779b650
commit bb27e37710
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/feature/tests/test_peak.py
+10 -2
View File
@@ -51,15 +51,23 @@ def test_flat_peak():
def test_num_peaks():
image = np.zeros((3, 7), dtype=np.uint8)
image = np.zeros((7, 7), dtype=np.uint8)
image[1, 1] = 10
image[1, 3] = 11
image[1, 5] = 12
assert len(peak.peak_local_max(image, min_distance=1)) == 3
image[3, 5] = 8
image[5, 3] = 7
assert len(peak.peak_local_max(image, min_distance=1)) == 5
peaks_limited = peak.peak_local_max(image, min_distance=1, num_peaks=2)
assert len(peaks_limited) == 2
assert (1, 3) in peaks_limited
assert (1, 5) in peaks_limited
peaks_limited = peak.peak_local_max(image, min_distance=1, num_peaks=4)
assert len(peaks_limited) == 4
assert (1, 3) in peaks_limited
assert (1, 5) in peaks_limited
assert (1, 1) in peaks_limited
assert (3, 5) in peaks_limited
if __name__ == '__main__':