25 lines
664 B
Python
25 lines
664 B
Python
|
"""Tests for parabolic cylinder functions.
|
||
|
|
|
||
|
|
"""
|
||
|
|
import numpy as np
|
||
|
|
from numpy.testing import assert_allclose, assert_equal
|
||
|
|
import scipy.special as sc
|
||
|
|
|
||
|
|
|
||
|
|
def test_pbwa_segfault():
|
||
|
|
# Regression test for https://github.com/scipy/scipy/issues/6208.
|
||
|
|
#
|
||
|
|
# Data generated by mpmath.
|
||
|
|
#
|
||
|
|
w = 1.02276567211316867161
|
||
|
|
wp = -0.48887053372346189882
|
||
|
|
assert_allclose(sc.pbwa(0, 0), (w, wp), rtol=1e-13, atol=0)
|
||
|
|
|
||
|
|
|
||
|
|
def test_pbwa_nan():
|
||
|
|
# Check that NaN's are returned outside of the range in which the
|
||
|
|
# implementation is accurate.
|
||
|
|
pts = [(-6, -6), (-6, 6), (6, -6), (6, 6)]
|
||
|
|
for p in pts:
|
||
|
|
assert_equal(sc.pbwa(*p), (np.nan, np.nan))
|