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
|
#!/usr/bin/python
# -*- coding: utf-8 -*-
"""test strategies for propagating values to the next level"""
import numpy as np
from multipletau import autocorrelate, correlate
def test_ac_compress_average():
ist = autocorrelate(range(42), m=2, dtype=np.float_, compress="average")
soll = np.array([[0.00000000e+00, 2.38210000e+04],
[1.00000000e+00, 2.29600000e+04],
[2.00000000e+00, 2.21000000e+04],
[4.00000000e+00, 2.03775000e+04],
[8.00000000e+00, 1.50612000e+04]])
assert np.allclose(soll, ist)
def test_cc_compress_average():
ist = correlate(range(42), range(1, 43), m=2, dtype=np.float_,
compress="average")
soll = np.array([[0.00000000e+00, 2.46820000e+04],
[1.00000000e+00, 2.38210000e+04],
[2.00000000e+00, 2.29600000e+04],
[4.00000000e+00, 2.12325000e+04],
[8.00000000e+00, 1.58508000e+04]])
assert np.allclose(soll, ist)
def test_ac_compress_first():
ist = autocorrelate(range(42), m=2, dtype=np.float_,
compress="first")
soll = np.array([[0.00000e+00, 2.38210e+04],
[1.00000e+00, 2.29600e+04],
[2.00000e+00, 2.21000e+04],
[4.00000e+00, 1.96080e+04],
[8.00000e+00, 1.31712e+04]])
assert np.allclose(soll, ist)
def test_cc_compress_first():
ist = correlate(range(42), range(1, 43), m=2, dtype=np.float_,
compress="first")
soll = np.array([[0.00000e+00, 2.46820e+04],
[1.00000e+00, 2.38210e+04],
[2.00000e+00, 2.29600e+04],
[4.00000e+00, 2.04440e+04],
[8.00000e+00, 1.39104e+04]])
assert np.allclose(soll, ist)
def test_ac_compress_second():
ist = autocorrelate(range(42), m=2, dtype=np.float_,
compress="second")
soll = np.array([[0.00000e+00, 2.38210e+04],
[1.00000e+00, 2.29600e+04],
[2.00000e+00, 2.21000e+04],
[4.00000e+00, 2.11660e+04],
[8.00000e+00, 1.71024e+04]])
assert np.allclose(soll, ist)
def test_cc_compress_second():
ist = correlate(range(42), range(1, 43), m=2, dtype=np.float_,
compress="second")
soll = np.array([[0.00000e+00, 2.46820e+04],
[1.00000e+00, 2.38210e+04],
[2.00000e+00, 2.29600e+04],
[4.00000e+00, 2.20400e+04],
[8.00000e+00, 1.79424e+04]])
assert np.allclose(soll, ist)
if __name__ == "__main__":
# Run all tests
loc = locals()
for key in list(loc.keys()):
if key.startswith("test_") and hasattr(loc[key], "__call__"):
loc[key]()
|
