46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import numpy as np
|
|
from numpy import pi, log, sqrt
|
|
from numpy.testing import assert_, assert_equal
|
|
|
|
from scipy.special._testutils import FuncData
|
|
import scipy.special as sc
|
|
|
|
# Euler-Mascheroni constant
|
|
euler = 0.57721566490153286
|
|
|
|
|
|
def test_consistency():
|
|
# Make sure the implementation of digamma for real arguments
|
|
# agrees with the implementation of digamma for complex arguments.
|
|
|
|
# It's all poles after -1e16
|
|
x = np.r_[-np.logspace(15, -30, 200), np.logspace(-30, 300, 200)]
|
|
dataset = np.vstack((x + 0j, sc.digamma(x))).T
|
|
FuncData(sc.digamma, dataset, 0, 1, rtol=5e-14, nan_ok=True).check()
|
|
|
|
|
|
def test_special_values():
|
|
# Test special values from Gauss's digamma theorem. See
|
|
#
|
|
# https://en.wikipedia.org/wiki/Digamma_function
|
|
|
|
dataset = [
|
|
(1, -euler),
|
|
(0.5, -2*log(2) - euler),
|
|
(1/3, -pi/(2*sqrt(3)) - 3*log(3)/2 - euler),
|
|
(1/4, -pi/2 - 3*log(2) - euler),
|
|
(1/6, -pi*sqrt(3)/2 - 2*log(2) - 3*log(3)/2 - euler),
|
|
(1/8,
|
|
-pi/2 - 4*log(2) - (pi + log(2 + sqrt(2)) - log(2 - sqrt(2)))/sqrt(2) - euler)
|
|
]
|
|
|
|
dataset = np.asarray(dataset)
|
|
FuncData(sc.digamma, dataset, 0, 1, rtol=1e-14).check()
|
|
|
|
|
|
def test_nonfinite():
|
|
pts = [0.0, -0.0, np.inf]
|
|
std = [-np.inf, np.inf, np.inf]
|
|
assert_equal(sc.digamma(pts), std)
|
|
assert_(all(np.isnan(sc.digamma([-np.inf, -1]))))
|