summaryrefslogtreecommitdiff
path: root/numpy/lib/tests/test_index_tricks.py
diff options
context:
space:
mode:
authorSebastian Berg <sebastian@sipsolutions.net>2013-09-09 22:41:03 +0200
committerSebastian Berg <sebastian@sipsolutions.net>2013-09-09 22:43:15 +0200
commit906a8848a3ea7eb0c5fa85653640b37caa08bbcd (patch)
tree9ef2e4de48c67a0084fd5a627a8d02d8ae7c27ef /numpy/lib/tests/test_index_tricks.py
parentc9b06111227f7a4ec213571f97e1b8d19b9c23f5 (diff)
downloadnumpy-906a8848a3ea7eb0c5fa85653640b37caa08bbcd.tar.gz
BUG: Fix 0-sized ndindex
The zerosize_ok flag to nditer was missing, so that it did not allow for 0-sized iteration. Closes gh-3714
Diffstat (limited to 'numpy/lib/tests/test_index_tricks.py')
-rw-r--r--numpy/lib/tests/test_index_tricks.py4
1 files changed, 4 insertions, 0 deletions
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()