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
|
import numpy as N
from numpy.testing import *
set_package_path()
from scipy.signal import wavelets
restore_path()
class test_wavelets(NumpyTestCase):
def check_qmf(self):
assert_array_equal(wavelets.qmf([1,1]),[1,-1])
def check_daub(self):
for i in xrange(1,15):
assert_equal(len(wavelets.daub(i)),i*2)
def check_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 check_morlet(self):
x = wavelets.morlet(50,4.1,complete=True)
y = wavelets.morlet(50,4.1,complete=False)
assert_equal(len(x),len(y))
x = wavelets.morlet(10,50,complete=False)
y = wavelets.morlet(10,50,complete=True)
assert_equal(x,y)
if __name__ == "__main__":
NumpyTest().run()
|
