Implement new assert_nD utility function

This commit is contained in:
Steven Silvester
2014-09-20 15:43:00 -05:00
parent 525a5f80c6
commit d6c64997cb
2 changed files with 20 additions and 17 deletions
+8 -1
View File
@@ -8,7 +8,7 @@ import six
from ._warnings import all_warnings
__all__ = ['deprecated', 'get_bound_method_class', 'all_warnings',
'safe_as_int']
'safe_as_int', 'assert_nD']
class skimage_deprecation(Warning):
@@ -141,3 +141,10 @@ def safe_as_int(val, atol=1e-3):
"{0}, check inputs.".format(val))
return np.round(val).astype(np.int64)
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"
raise ValueError(msg % (arg_name, ndim))