diff options
| author | Ralf Gommers <ralf.gommers@googlemail.com> | 2011-12-28 10:01:42 +0100 |
|---|---|---|
| committer | Ralf Gommers <ralf.gommers@googlemail.com> | 2011-12-28 10:01:42 +0100 |
| commit | 4e571172a210612bdeba1db0135b7d5fbc44ee0c (patch) | |
| tree | ac5862fc0e228b85b69aeabb2d0d34581456b177 /numpy/lib/function_base.py | |
| parent | 37d723cb629e3e8cb4d9ed7b85702fbd6350818d (diff) | |
| download | numpy-4e571172a210612bdeba1db0135b7d5fbc44ee0c.tar.gz | |
BUG: meshgrid: raise error on single input.
Diffstat (limited to 'numpy/lib/function_base.py')
| -rw-r--r-- | numpy/lib/function_base.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index abb239ffc..8e2fc375e 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -3288,13 +3288,14 @@ def meshgrid(*xi, **kwargs): """ copy_ = kwargs.get('copy', True) + + if len(xi) < 2: + msg = 'meshgrid() takes 2 or more arguments (%d given)' % int(len(xi) > 0) + raise ValueError(msg) + args = np.atleast_1d(*xi) ndim = len(args) - if ndim < 2: - msg = 'meshgrid() takes 2 or more arguments (%d given)' % int(ndim > 0) - raise TypeError(msg) - sparse = kwargs.get('sparse', False) indexing = kwargs.get('indexing', 'xy') |