diff options
| author | David Cournapeau <cournape@gmail.com> | 2010-03-31 02:26:14 +0000 |
|---|---|---|
| committer | David Cournapeau <cournape@gmail.com> | 2010-03-31 02:26:14 +0000 |
| commit | ced34d27a8eef42e0f963afa4989e2383fc3ca77 (patch) | |
| tree | 0b1d244cb0d4c0bce10b0206c4056cc5e91d9b8d /numpy | |
| parent | cfb3c55e3813cc05673a37452dea405fc0acea03 (diff) | |
| download | numpy-ced34d27a8eef42e0f963afa4989e2383fc3ca77.tar.gz | |
BUG: fix kaiser for M=1.
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/lib/function_base.py | 2 | ||||
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 2 |
2 files changed, 3 insertions, 1 deletions
diff --git a/numpy/lib/function_base.py b/numpy/lib/function_base.py index 87ecf717e..3b8ddf099 100644 --- a/numpy/lib/function_base.py +++ b/numpy/lib/function_base.py @@ -2608,6 +2608,8 @@ def kaiser(M,beta): """ from numpy.dual import i0 + if M == 1: + return np.array([1.]) n = arange(0,M) alpha = (M-1)/2.0 return i0(beta * sqrt(1-((n-alpha)/alpha)**2.0))/i0(float(beta)) diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 36db8baa4..330a1a51f 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -778,7 +778,7 @@ class Test_i0(TestCase): class TestKaiser(TestCase): def test_simple(self): assert_almost_equal(kaiser(0, 1.0), array([])) - assert isnan(kaiser(1, 1.0)) + assert isfinite(kaiser(1, 1.0)) assert_almost_equal(kaiser(2, 1.0), array([ 0.78984831, 0.78984831])) assert_almost_equal(kaiser(5, 1.0), array([ 0.78984831, 0.94503323, 1. , |