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

Allow assert_nD to take multiple dimensions

Browse Source
This commit is contained in:
Steven Silvester
2014-09-20 17:12:35 -05:00
parent 6a8c5e460f
commit 17ff39ff25
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/_shared/utils.py
+5 -2
View File
@@ -145,6 +145,9 @@ def safe_as_int(val, atol=1e-3):
def assert_nD(array, arg_name='image', ndim=2):
array = np.asanyarray(array)
if array.ndim != ndim:
msg = "The parameter `%s` must be a %s-dimensional array"
if isinstance(ndim, int):
ndim = [ndim]
if not array.ndim in ndim:
msg = "The parameter `%s` must be a %s-dimensional array"
raise ValueError(msg % (arg_name, ndim))
raise ValueError(msg % (arg_name, '-or-'.join([str(n) for n in ndim]))