diff options
| author | jaimefrio <jaime.frio@gmail.com> | 2014-02-09 22:06:13 -0800 |
|---|---|---|
| committer | jaimefrio <jaime.frio@gmail.com> | 2014-02-09 22:11:27 -0800 |
| commit | f65593dd69cae9b0fb30c06c07868dafb8436d11 (patch) | |
| tree | 462bfb0315d399e374d4721154712e8cc50f1087 /numpy/lib/tests/test_function_base.py | |
| parent | 839225599b1fb18ebb09424459ebd2066a8f21f9 (diff) | |
| download | numpy-f65593dd69cae9b0fb30c06c07868dafb8436d11.tar.gz | |
STY: adapted code to `C_STYLE_GUIDE.rst.txt`.
MAINT: rewrote `check_array_monotonic` to use array indices, not pointers.
TST: tests for proper handling of bins with all items almost equal now check
the return value for correctness, not just that an error is not raised.
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 0d2a66f53..0662d7089 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -760,15 +760,18 @@ class TestDigitize(TestCase): x = rand(10) bins = np.linspace(x.min(), x.max(), 10) assert_(np.all(digitize(x, bins, True) != 10)) - + def test_monotonic(self): - x = [0] + x = [-1, 0, 1, 2] bins = [0, 0, 1] - digitize(x, bins) + 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] - digitize(x, bins) + 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] - digitize(x, bins) + 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] |