diff options
| author | peterbell10 <peterbell10@live.co.uk> | 2020-08-12 07:36:07 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-08-12 09:36:07 +0300 |
| commit | a2b9c2d5b6637b040917c0a2ef393dae83f09ee3 (patch) | |
| tree | afa02fc2d1bdf8115d4abf49aa2f99db02094841 /numpy/lib/tests/test_function_base.py | |
| parent | 7ec2e1bac72afcdc68cf8256879afbc4cb14a907 (diff) | |
| download | numpy-a2b9c2d5b6637b040917c0a2ef393dae83f09ee3.tar.gz | |
API, BUG: Raise error on complex input to i0 (#17062)
* BUG, API: Raise error on complex input to np.i0
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 89c1a2d9b..635fe1432 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -2111,8 +2111,9 @@ class Test_I0: i0(0.5), np.array(1.0634833707413234)) - A = np.array([0.49842636, 0.6969809, 0.22011976, 0.0155549]) - expected = np.array([1.06307822, 1.12518299, 1.01214991, 1.00006049]) + # need at least one test above 8, as the implementation is piecewise + A = np.array([0.49842636, 0.6969809, 0.22011976, 0.0155549, 10.0]) + expected = np.array([1.06307822, 1.12518299, 1.01214991, 1.00006049, 2815.71662847]) assert_almost_equal(i0(A), expected) assert_almost_equal(i0(-A), expected) @@ -2149,6 +2150,10 @@ class Test_I0: assert_array_equal(exp, res) + def test_complex(self): + a = np.array([0, 1 + 2j]) + with pytest.raises(TypeError, match="i0 not supported for complex values"): + res = i0(a) class TestKaiser: |