Vehicle-cpp/hkpc/software/scipy/special/tests/test_pcf.py

"""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))