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
|
import numpy as np
from numpy.testing import TestCase, run_module_suite, assert_equal, \
assert_array_equal, assert_array_almost_equal, assert_array_less, assert_
from scipy.signal import wavelets
class TestWavelets(TestCase):
def test_qmf(self):
assert_array_equal(wavelets.qmf([1,1]),[1,-1])
def test_daub(self):
for i in xrange(1,15):
assert_equal(len(wavelets.daub(i)),i*2)
def test_cascade(self):
for J in xrange(1,7):
for i in xrange(1,5):
lpcoef = wavelets.daub(i)
k = len(lpcoef)
x,phi,psi = wavelets.cascade(lpcoef,J)
assert_(len(x) == len(phi) == len(psi))
assert_equal(len(x),(k-1)*2**J)
def test_morlet(self):
x = wavelets.morlet(50,4.1,complete=True)
y = wavelets.morlet(50,4.1,complete=False)
# Test if complete and incomplete wavelet have same lengths:
assert_equal(len(x),len(y))
# Test if complete wavelet is less than incomplete wavelet:
assert_array_less(x,y)
x = wavelets.morlet(10,50,complete=False)
y = wavelets.morlet(10,50,complete=True)
# For large widths complete and incomplete wavelets should be
# identical within numerical precision:
assert_equal(x,y)
# miscellaneous tests:
x = np.array([1.73752399e-09 +9.84327394e-25j,
6.49471756e-01 +0.00000000e+00j,
1.73752399e-09 -9.84327394e-25j])
y = wavelets.morlet(3,w=2,complete=True)
assert_array_almost_equal(x,y)
x = np.array([2.00947715e-09 +9.84327394e-25j,
7.51125544e-01 +0.00000000e+00j,
2.00947715e-09 -9.84327394e-25j])
y = wavelets.morlet(3,w=2,complete=False)
assert_array_almost_equal(x,y,decimal=2)
x = wavelets.morlet(10000,s=4,complete=True)
y = wavelets.morlet(20000,s=8,complete=True)[5000:15000]
assert_array_almost_equal(x,y,decimal=2)
x = wavelets.morlet(10000,s=4,complete=False)
assert_array_almost_equal(y,x,decimal=2)
y = wavelets.morlet(20000,s=8,complete=False)[5000:15000]
assert_array_almost_equal(x,y,decimal=2)
x = wavelets.morlet(10000,w=3,s=5,complete=True)
y = wavelets.morlet(20000,w=3,s=10,complete=True)[5000:15000]
assert_array_almost_equal(x,y,decimal=2)
x = wavelets.morlet(10000,w=3,s=5,complete=False)
assert_array_almost_equal(y,x,decimal=2)
y = wavelets.morlet(20000,w=3,s=10,complete=False)[5000:15000]
assert_array_almost_equal(x,y,decimal=2)
x = wavelets.morlet(10000,w=7,s=10,complete=True)
y = wavelets.morlet(20000,w=7,s=20,complete=True)[5000:15000]
assert_array_almost_equal(x,y,decimal=2)
x = wavelets.morlet(10000,w=7,s=10,complete=False)
assert_array_almost_equal(x,y,decimal=2)
y = wavelets.morlet(20000,w=7,s=20,complete=False)[5000:15000]
assert_array_almost_equal(x,y,decimal=2)
if __name__ == "__main__":
run_module_suite()
|
