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

Add test coverage for guess_spatial_dimensions()

Browse Source
This commit is contained in:
Juan Nunez-Iglesias
2013-07-13 01:27:17 +02:00
parent f0ecc4d00d
commit 6221ce6e3b
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/color/tests/test_colorconv.py
+15 -1
View File
@@ -33,7 +33,8 @@ from skimage.color import (rgb2hsv, hsv2rgb,
rgb2grey, gray2rgb,
xyz2lab, lab2xyz,
lab2rgb, rgb2lab,
is_rgb, is_gray
is_rgb, is_gray,
guess_spatial_dimensions
)
from skimage import data_dir, data
@@ -41,6 +42,19 @@ from skimage import data_dir, data
import colorsys
def test_guess_spatial_dimensions():
im1 = np.zeros((5, 5))
im2 = np.zeros((5, 5, 5))
im3 = np.zeros((5, 5, 3))
im4 = np.zeros((5, 5, 5, 3))
im5 = np.zeros((5,))
assert_equal(guess_spatial_dimensions(im1), 2)
assert_equal(guess_spatial_dimensions(im2), 3)
assert_equal(guess_spatial_dimensions(im3), None)
assert_equal(guess_spatial_dimensions(im4), 3)
assert_raises(guess_spatial_dimensions(im5), ValueError)
class TestColorconv(TestCase):
img_rgb = imread(os.path.join(data_dir, 'color.png'))