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

Add test case for different dtypes of structuring element

Browse Source
This commit is contained in:
Johannes Schönberger
2013-03-27 22:26:29 +01:00
parent 23cf8b7dfb
commit 8299f471f2
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/filter/rank/tests/test_rank.py
+20
View File
@@ -376,5 +376,25 @@ def test_entropy():
assert(np.max(rank.entropy(data, selem)) == 12000)
def test_selem_dtypes():
image = np.zeros((5, 5), dtype=np.uint8)
out = np.zeros_like(image)
mask = np.ones_like(image, dtype=np.uint8)
image[2, 2] = 255
image[2, 3] = 128
image[1, 2] = 16
for dtype in (np.uint8, np.uint16, np.int32, np.int64,
np.float32, np.float64):
elem = np.array([[0, 0, 0], [0, 1, 0], [0, 0, 0]], dtype=dtype)
rank.mean(image=image, selem=elem, out=out, mask=mask,
shift_x=0, shift_y=0)
assert_array_equal(image, out)
rank.percentile_mean(image=image, selem=elem, out=out, mask=mask,
shift_x=0, shift_y=0)
assert_array_equal(image, out)
if __name__ == "__main__":
run_module_suite()