diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2014-02-11 18:16:02 -0700 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2014-02-11 18:16:02 -0700 |
| commit | 58ffcdc62b1db9d6c762a82ca483484a5cd7c92b (patch) | |
| tree | cda9017ae4d767a97d6e1e3e307b1e6c3af5c1aa /numpy/lib/tests/test_function_base.py | |
| parent | e2addbd77fbed5aa64a07f9b08e217c63e15467f (diff) | |
| parent | c8975ff262292d760b533bb433de50b686b15f9e (diff) | |
| download | numpy-58ffcdc62b1db9d6c762a82ca483484a5cd7c92b.tar.gz | |
Merge pull request #4247 from jaimefrio/digitize-monotonic
BUG: check for monotonic bin arrays in digitize
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 4d2360b69..003d3e541 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -761,6 +761,22 @@ class TestDigitize(TestCase): bins = np.linspace(x.min(), x.max(), 10) assert_(np.all(digitize(x, bins, True) != 10)) + def test_monotonic(self): + x = [-1, 0, 1, 2] + bins = [0, 0, 1] + assert_array_equal(digitize(x, bins, False), [0, 2, 3, 3]) + assert_array_equal(digitize(x, bins, True), [0, 0, 2, 3]) + bins = [1, 1, 0] + assert_array_equal(digitize(x, bins, False), [3, 2, 0, 0]) + assert_array_equal(digitize(x, bins, True), [3, 3, 2, 0]) + 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]) + bins = [0, 0, 1, 0] + assert_raises(ValueError, digitize, x, bins) + bins = [1, 1, 0, 1] + assert_raises(ValueError, digitize, x, bins) + class TestUnwrap(TestCase): def test_simple(self): |