diff options
Diffstat (limited to 'numpy/lib')
-rw-r--r-- | numpy/lib/index_tricks.py | 3 | ||||
-rw-r--r-- | numpy/lib/tests/test_index_tricks.py | 4 |
2 files changed, 6 insertions, 1 deletions
diff --git a/numpy/lib/index_tricks.py b/numpy/lib/index_tricks.py index 570cd0f1d..f0066be81 100644 --- a/numpy/lib/index_tricks.py +++ b/numpy/lib/index_tricks.py @@ -540,7 +540,8 @@ class ndindex(object): if len(shape) == 1 and isinstance(shape[0], tuple): shape = shape[0] x = as_strided(_nx.zeros(1), shape=shape, strides=_nx.zeros_like(shape)) - self._it = _nx.nditer(x, flags=['multi_index'], order='C') + self._it = _nx.nditer(x, flags=['multi_index', 'zerosize_ok'], + order='C') def __iter__(self): return self diff --git a/numpy/lib/tests/test_index_tricks.py b/numpy/lib/tests/test_index_tricks.py index b4152fafa..6b01464a7 100644 --- a/numpy/lib/tests/test_index_tricks.py +++ b/numpy/lib/tests/test_index_tricks.py @@ -274,6 +274,10 @@ def test_ndindex(): x = list(np.ndindex(())) assert_equal(x, [()]) + # Make sure 0-sized ndindex works correctly + x = list(np.ndindex(*[0])) + assert_equal(x, []) + if __name__ == "__main__": run_module_suite() |