diff options
author | Alex <aprockhill206@gmail.com> | 2020-05-15 09:56:38 -0700 |
---|---|---|
committer | Alex <aprockhill206@gmail.com> | 2020-07-22 11:59:31 -0700 |
commit | 5840165bd7db8628d0d5b318544943a28d799068 (patch) | |
tree | edb285c83873da369951dc33a5f38f1f02a82054 /numpy/lib/tests/test_function_base.py | |
parent | 8c4ce9936ec6989fdf2b2374489f023681e329a3 (diff) | |
download | numpy-5840165bd7db8628d0d5b318544943a28d799068.tar.gz |
simplified
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
-rw-r--r-- | numpy/lib/tests/test_function_base.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index d6e768a2b..32f660772 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -1715,6 +1715,9 @@ class TestDigitize: bins = [-1, 0, 1, 2] assert_array_equal(digitize(x, bins, False, True), [1, 2, 3, 3]) assert_array_equal(digitize(x, bins, True, True), [1, 1, 2, 3]) + bins = [2, 1, 0, -1] + assert_array_equal(digitize(x, bins, False, True), [3, 2, 1, 1]) + assert_array_equal(digitize(x, bins, True, True), [3, 3, 2, 1]) bins = [1, 1, 1, 1] assert_array_equal(digitize(x, bins, False), [0, 0, 4, 4]) assert_array_equal(digitize(x, bins, True), [0, 0, 0, 4]) |