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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
# Test the functions in galpy/util/__init__.py
import numpy
def test_save_pickles():
import os
import pickle
import tempfile
from galpy.util import save_pickles
savethis = numpy.linspace(0.0, 100.0, 1001)
savefile, tmp_savefilename = tempfile.mkstemp()
try:
os.close(savefile) # Easier this way
save_pickles(tmp_savefilename, savethis)
savefile = open(tmp_savefilename, "rb")
restorethis = pickle.load(savefile)
savefile.close()
assert numpy.all(numpy.fabs(restorethis - savethis) < 10.0**-10.0), (
"save_pickles did not work as expected"
)
finally:
os.remove(tmp_savefilename)
# Also test the handling of KeyboardInterrupt
try:
save_pickles(tmp_savefilename, savethis, testKeyboardInterrupt=True)
except KeyboardInterrupt:
pass
else:
raise AssertionError(
"save_pickles with testKeyboardInterrupt=True did not raise KeyboardInterrupt"
)
savefile = open(tmp_savefilename, "rb")
restorethis = pickle.load(savefile)
savefile.close()
assert numpy.all(numpy.fabs(restorethis - savethis) < 10.0**-10.0), (
"save_pickles did not work as expected when KeyboardInterrupted"
)
if os.path.exists(tmp_savefilename):
os.remove(tmp_savefilename)
return None
def test_logsumexp():
from galpy.util import logsumexp
sumthis = numpy.array([[0.0, 1.0]])
sum = numpy.log(numpy.exp(0.0) + numpy.exp(1.0))
assert numpy.all(numpy.fabs(logsumexp(sumthis, axis=0) - sumthis) < 10.0**-10.0), (
"galpy.util.logsumexp did not work as expected"
)
assert numpy.fabs(logsumexp(sumthis, axis=1) - sum) < 10.0**-10.0, (
"galpy.util.logsumexp did not work as expected"
)
assert numpy.fabs(logsumexp(sumthis, axis=None) - sum) < 10.0**-10.0, (
"galpy.util.logsumexp did not work as expected"
)
return None
def test_fast_cholesky_invert():
from galpy.util import fast_cholesky_invert
matrix = numpy.array([[2.0, 1.0], [1.0, 4.0]])
invmatrix = fast_cholesky_invert(matrix)
unit = numpy.dot(invmatrix, matrix)
assert numpy.all(numpy.fabs(numpy.diag(unit) - 1.0) < 10.0**-8.0), (
"fast_cholesky_invert did not work as expected"
)
assert numpy.fabs(unit[0, 1] - 0.0) < 10.0**-8.0, (
"fast_cholesky_invert did not work as expected"
)
assert numpy.fabs(unit[1, 0] - 0.0) < 10.0**-8.0, (
"fast_cholesky_invert did not work as expected"
)
# Check the other way around
unit = numpy.dot(matrix, invmatrix)
assert numpy.all(numpy.fabs(numpy.diag(unit) - 1.0) < 10.0**-8.0), (
"fast_cholesky_invert did not work as expected"
)
assert numpy.fabs(unit[0, 1] - 0.0) < 10.0**-8.0, (
"fast_cholesky_invert did not work as expected"
)
assert numpy.fabs(unit[1, 0] - 0.0) < 10.0**-8.0, (
"fast_cholesky_invert did not work as expected"
)
# Also check determinant
invmatrix, logdet = fast_cholesky_invert(matrix, logdet=True)
assert numpy.fabs(logdet - numpy.log(7.0)) < 10.0**-8.0, (
"fast_cholesky_invert's determinant did not work as expected"
)
return None
def test_quadpack():
from galpy.util.quadpack import dblquad
int = dblquad(lambda y, x: 4.0 * x * y, 0.0, 1.0, lambda z: 0.0, lambda z: 1.0)
assert numpy.fabs(int[0] - 1.0) < int[1], (
"galpy.util.quadpack.dblquad did not work as expected"
)
return None
|
