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
|
import numpy as N
from numpy.testing import *
set_package_path()
from scipy.special import gammaln, multigammaln
restore_path()
class test_multigammaln(NumpyTestCase):
def test1(self):
a = N.abs(N.random.randn())
assert_array_equal(multigammaln(a, 1), gammaln(a))
def test_ararg(self):
d = 5
a = N.abs(N.random.randn(3, 2)) + d
tr = multigammaln(a, d)
assert_array_equal(tr.shape, a.shape)
for i in range(a.size):
assert_array_equal(tr.ravel()[i], multigammaln(a.ravel()[i], d))
d = 5
a = N.abs(N.random.randn(1, 2)) + d
tr = multigammaln(a, d)
assert_array_equal(tr.shape, a.shape)
for i in range(a.size):
assert_array_equal(tr.ravel()[i], multigammaln(a.ravel()[i], d))
def test_bararg(self):
try:
multigammaln(0.5, 1.2)
raise Exception("Expected this call to fail")
except ValueError:
pass
if __name__ == '__main__':
NumpyTest().run()
|
