37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
import numpy as np
|
||
|
|
|
||
|
|
import scipy.special as sc
|
||
|
|
from scipy.special._testutils import FuncData
|
||
|
|
|
||
|
|
|
||
|
|
def test_sici_consistency():
|
||
|
|
# Make sure the implementation of sici for real arguments agrees
|
||
|
|
# with the implementation of sici for complex arguments.
|
||
|
|
|
||
|
|
# On the negative real axis Cephes drops the imaginary part in ci
|
||
|
|
def sici(x):
|
||
|
|
si, ci = sc.sici(x + 0j)
|
||
|
|
return si.real, ci.real
|
||
|
|
|
||
|
|
x = np.r_[-np.logspace(8, -30, 200), 0, np.logspace(-30, 8, 200)]
|
||
|
|
si, ci = sc.sici(x)
|
||
|
|
dataset = np.column_stack((x, si, ci))
|
||
|
|
FuncData(sici, dataset, 0, (1, 2), rtol=1e-12).check()
|
||
|
|
|
||
|
|
|
||
|
|
def test_shichi_consistency():
|
||
|
|
# Make sure the implementation of shichi for real arguments agrees
|
||
|
|
# with the implementation of shichi for complex arguments.
|
||
|
|
|
||
|
|
# On the negative real axis Cephes drops the imaginary part in chi
|
||
|
|
def shichi(x):
|
||
|
|
shi, chi = sc.shichi(x + 0j)
|
||
|
|
return shi.real, chi.real
|
||
|
|
|
||
|
|
# Overflow happens quickly, so limit range
|
||
|
|
x = np.r_[-np.logspace(np.log10(700), -30, 200), 0,
|
||
|
|
np.logspace(-30, np.log10(700), 200)]
|
||
|
|
shi, chi = sc.shichi(x)
|
||
|
|
dataset = np.column_stack((x, shi, chi))
|
||
|
|
FuncData(shichi, dataset, 0, (1, 2), rtol=1e-14).check()
|