diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2015-09-27 12:33:09 -0600 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2015-09-27 12:33:09 -0600 |
| commit | ea289ee45dffa85d2fde07685eb3334816d8af7c (patch) | |
| tree | 92d94ad578e6d2ba26def28837725f6cf0e5690e /numpy/lib/tests | |
| parent | 41afcc3681d250f231aea9d9f428a9e197a47f6e (diff) | |
| parent | f29c387272a9279f82ab04bbbe1bb68040b6d383 (diff) | |
| download | numpy-ea289ee45dffa85d2fde07685eb3334816d8af7c.tar.gz | |
Merge pull request #6371 from seberg/pr-5771
BUG: Make sure warning for array split is always applied
Diffstat (limited to 'numpy/lib/tests')
| -rw-r--r-- | numpy/lib/tests/test_shape_base.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_shape_base.py b/numpy/lib/tests/test_shape_base.py index 3f2d8d5b4..8ab72b9f9 100644 --- a/numpy/lib/tests/test_shape_base.py +++ b/numpy/lib/tests/test_shape_base.py @@ -111,6 +111,15 @@ class TestArraySplit(TestCase): compare_results(res, desired) assert_(a.dtype.type is res[-1].dtype.type) + # Same thing for manual splits: + res = assert_warns(FutureWarning, array_split, a, [0, 1, 2], axis=0) + + # After removing the FutureWarning, the last should be zeros((0, 10)) + desired = [np.array([]), np.array([np.arange(10)]), + np.array([np.arange(10)])] + compare_results(res, desired) + assert_(a.dtype.type is res[-1].dtype.type) + def test_integer_split_2D_cols(self): a = np.array([np.arange(10), np.arange(10)]) res = array_split(a, 3, axis=-1) |