diff options
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/lib/tests/test_index_tricks.py | 3 | ||||
| -rw-r--r-- | numpy/lib/tests/test_twodim_base.py | 2 |
2 files changed, 3 insertions, 2 deletions
diff --git a/numpy/lib/tests/test_index_tricks.py b/numpy/lib/tests/test_index_tricks.py index 50dc5ac15..afa94396a 100644 --- a/numpy/lib/tests/test_index_tricks.py +++ b/numpy/lib/tests/test_index_tricks.py @@ -1,6 +1,7 @@ from numpy.testing import * from numpy import ( array, ones, r_, mgrid, unravel_index, zeros, where, - fill_diagonal, diag_indices, diag_indices_from ) + ndenumerate, fill_diagonal, diag_indices, + diag_indices_from ) class TestUnravelIndex(TestCase): def test_basic(self): diff --git a/numpy/lib/tests/test_twodim_base.py b/numpy/lib/tests/test_twodim_base.py index a1e58d53b..45ab54fbc 100644 --- a/numpy/lib/tests/test_twodim_base.py +++ b/numpy/lib/tests/test_twodim_base.py @@ -248,7 +248,7 @@ def test_triu_indices(): # Both for indexing: yield (assert_array_equal, a[iu1], - array([ 1, 5, 6, 9, 10, 11, 13, 14, 15, 16]) ) + array([1, 2, 3, 4, 6, 7, 8, 11, 12, 16])) # And for assigning values: a[iu1] = -1 |