ai-content-maker/.venv/Lib/site-packages/scipy/special/tests/test_powm1.py

66 lines
2.2 KiB
Python
Raw Normal View History

2024-05-03 04:18:51 +03:00
import pytest
import numpy as np
from numpy.testing import assert_allclose
from scipy.special import powm1
# Expected values were computed with mpmath, e.g.
#
# >>> import mpmath
# >>> mpmath.np.dps = 200
# >>> print(float(mpmath.powm1(2.0, 1e-7))
# 6.931472045825965e-08
#
powm1_test_cases = [
(1.25, 0.75, 0.18217701125396976, 1e-15),
(2.0, 1e-7, 6.931472045825965e-08, 1e-15),
(25.0, 5e-11, 1.6094379125636148e-10, 1e-15),
(0.99996, 0.75, -3.0000150002530058e-05, 1e-15),
(0.9999999999990905, 20, -1.81898940353014e-11, 1e-15),
(-1.25, 751.0, -6.017550852453444e+72, 2e-15)
]
@pytest.mark.parametrize('x, y, expected, rtol', powm1_test_cases)
def test_powm1(x, y, expected, rtol):
p = powm1(x, y)
assert_allclose(p, expected, rtol=rtol)
@pytest.mark.parametrize('x, y, expected',
[(0.0, 0.0, 0.0),
(0.0, -1.5, np.inf),
(0.0, 1.75, -1.0),
(-1.5, 2.0, 1.25),
(-1.5, 3.0, -4.375),
(np.nan, 0.0, 0.0),
(1.0, np.nan, 0.0),
(1.0, np.inf, 0.0),
(1.0, -np.inf, 0.0),
(np.inf, 7.5, np.inf),
(np.inf, -7.5, -1.0),
(3.25, np.inf, np.inf),
(np.inf, np.inf, np.inf),
(np.inf, -np.inf, -1.0),
(np.inf, 0.0, 0.0),
(-np.inf, 0.0, 0.0),
(-np.inf, 2.0, np.inf),
(-np.inf, 3.0, -np.inf),
(-1.0, float(2**53 - 1), -2.0)])
def test_powm1_exact_cases(x, y, expected):
# Test cases where we have an exact expected value.
p = powm1(x, y)
assert p == expected
@pytest.mark.parametrize('x, y',
[(-1.25, 751.03),
(-1.25, np.inf),
(np.nan, np.nan),
(-np.inf, -np.inf),
(-np.inf, 2.5)])
def test_powm1_return_nan(x, y):
# Test cases where the expected return value is nan.
p = powm1(x, y)
assert np.isnan(p)