49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
|
import numpy as np
|
||
|
|
import scipy.special as sc
|
||
|
|
from numpy.testing import assert_almost_equal, assert_array_equal
|
||
|
|
|
||
|
|
|
||
|
|
class TestPdtr(object):
|
||
|
|
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(object):
|
||
|
|
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)
|