File size: 1,284 Bytes
7885a28 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
import numpy as np
import scipy.special as sc
from numpy.testing import assert_almost_equal, assert_array_equal
class TestPdtr:
def test(self):
val = sc.pdtr(0, 1)
assert_almost_equal(val, np.exp(-1))
def test_m_zero(self):
val = sc.pdtr([0, 1, 2], 0)
assert_array_equal(val, [1, 1, 1])
def test_rounding(self):
double_val = sc.pdtr([0.1, 1.1, 2.1], 1.0)
int_val = sc.pdtr([0, 1, 2], 1.0)
assert_array_equal(double_val, int_val)
def test_inf(self):
val = sc.pdtr(np.inf, 1.0)
assert_almost_equal(val, 1.0)
def test_domain(self):
val = sc.pdtr(-1.1, 1.0)
assert np.isnan(val)
class TestPdtrc:
def test_value(self):
val = sc.pdtrc(0, 1)
assert_almost_equal(val, 1 - np.exp(-1))
def test_m_zero(self):
val = sc.pdtrc([0, 1, 2], 0.0)
assert_array_equal(val, [0, 0, 0])
def test_rounding(self):
double_val = sc.pdtrc([0.1, 1.1, 2.1], 1.0)
int_val = sc.pdtrc([0, 1, 2], 1.0)
assert_array_equal(double_val, int_val)
def test_inf(self):
val = sc.pdtrc(np.inf, 1.0)
assert_almost_equal(val, 0.0)
def test_domain(self):
val = sc.pdtrc(-1.1, 1.0)
assert np.isnan(val)
|