def test_bior(self):
s1 = self.s1
x1 = self.x1
x2 = self.x2
wnames = ['bior1.1', 'bior1.3', 'bior1.5', 'bior2.2', 'bior2.4', 'bior2.6',
'bior2.8', 'bior3.1', 'bior3.3', 'bior3.5', 'bior3.7', 'bior3.9',
'bior4.4', 'bior5.5', 'bior6.8']
for N in np.arange(1,13):
wname = wnames[N]
[cA,cD] = pyyawt.dwt(x1,wname)
[Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname)
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x1,np.size(Lo_D),'b'),Lo_D),(np.size(x1)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x1,np.size(Lo_D),'b'),Hi_D),(np.size(x1)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(x2,wname)
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x2,np.size(Lo_D),'b'),Lo_D),(np.size(x2)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x2,np.size(Lo_D),'b'),Hi_D),(np.size(x2)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(s1,wname)
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',s1,np.size(Lo_D),'b'),Lo_D),(np.size(s1)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',s1,np.size(Lo_D),'b'),Hi_D),(np.size(s1)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
def test_haar(self):
s1 = self.s1
x1 = self.x1
x2 = self.x2
[cA,cD] = pyyawt.dwt(x1,'haar')
[Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters('haar')
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x1,np.size(Lo_D),'b'),Lo_D),(np.size(x1)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x1,np.size(Lo_D),'b'),Hi_D),(np.size(x1)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(x2,'haar')
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x2,np.size(Lo_D),'b'),Lo_D),(np.size(x2)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x2,np.size(Lo_D),'b'),Hi_D),(np.size(x2)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(s1,'haar')
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',s1,np.size(Lo_D),'b'),Lo_D),(np.size(s1)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',s1,np.size(Lo_D),'b'),Hi_D),(np.size(s1)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
def test_sym(self):
s1 = self.s1
x1 = self.x1
x2 = self.x2
for N in np.arange(4,11):
wname = "sym" + str(N)
[cA,cD] = pyyawt.dwt(x1,wname)
[Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname)
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x1,np.size(Lo_D),'b'),Lo_D),(np.size(x1)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x1,np.size(Lo_D),'b'),Hi_D),(np.size(x1)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(x2,wname)
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x2,np.size(Lo_D),'b'),Lo_D),(np.size(x2)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',x2,np.size(Lo_D),'b'),Hi_D),(np.size(x2)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(s1,wname)
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',s1,np.size(Lo_D),'b'),Lo_D),(np.size(s1)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,'symh',s1,np.size(Lo_D),'b'),Hi_D),(np.size(s1)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
def test_dwtMode(self):
s1 = self.s1
x1 = self.x1
x2 = self.x2
# ST = pyyawt.dwtmode("status","nodisp")
dwtModes = ['symw', 'asymh', 'asymw', 'zpd', 'sp0', 'sp1',
'ppd','per']
wname = "db10"
for N in np.arange(len(dwtModes)-1):
# pyyawt.dwtmode(dwtModes[N])
[cA,cD] = pyyawt.dwt(x1,wname,"mode",dwtModes[N])
[Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname)
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],x1,np.size(Lo_D),'b'),Lo_D),(np.size(x1)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],x1,np.size(Lo_D),'b'),Hi_D),(np.size(x1)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(x2,wname,"mode",dwtModes[N])
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],x2,np.size(Lo_D),'b'),Lo_D),(np.size(x2)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],x2,np.size(Lo_D),'b'),Hi_D),(np.size(x2)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(s1,wname,"mode",dwtModes[N])
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],s1,np.size(Lo_D),'b'),Lo_D),(np.size(s1)+np.size(Lo_D)-1)))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],s1,np.size(Lo_D),'b'),Hi_D),(np.size(s1)+np.size(Lo_D)-1)))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
N = len(dwtModes) - 1
[cA,cD] = pyyawt.dwt(x1,wname,"mode",dwtModes[N])
[Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname)
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],x1,np.size(Lo_D),'b'),Lo_D),(np.size(x1))))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],x1,np.size(Lo_D),'b'),Hi_D),(np.size(x1))))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(x2,wname,"mode",dwtModes[N])
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],x2,np.size(Lo_D),'b'),Lo_D),(np.size(x2))))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],x2,np.size(Lo_D),'b'),Hi_D),(np.size(x2))))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
[cA,cD] = pyyawt.dwt(s1,wname,"mode",dwtModes[N])
caa = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],s1,np.size(Lo_D),'b'),Lo_D),(np.size(s1))))
cdd = pyyawt.dyaddown(pyyawt.wkeep(pyyawt.conv(pyyawt.wextend(1,dwtModes[N],s1,np.size(Lo_D),'b'),Hi_D),(np.size(s1))))
np.testing.assert_almost_equal(caa.flatten(), cA)
np.testing.assert_almost_equal(cdd.flatten(), cD)
def test_vector(self):
a = np.random.randn(1,51)
b = pyyawt.wextend(1,'symh',a,7)
b1 = pyyawt.wextend(1,'symh',a,7,'b')
b2 = pyyawt.wextend(1,'symh',a,7,'l')
b3 = pyyawt.wextend(1,'symh',a,7,'r')
c = np.hstack((a[:,6::-1], a, a[:,-1:-8:-1]))
c1 = np.hstack((a[:,6::-1], a))
c2 = np.hstack((a, a[:,-1:-8:-1]))
np.testing.assert_almost_equal(b, c)
np.testing.assert_almost_equal(b1, c)
np.testing.assert_almost_equal(b2, c1)
np.testing.assert_almost_equal(b3, c2)
b = pyyawt.wextend(1,'sym',a,7)
b1 = pyyawt.wextend(1,'sym',a,7,'b')
b2 = pyyawt.wextend(1,'sym',a,7,'l')
b3 = pyyawt.wextend(1,'sym',a,7,'r')
c = np.hstack((a[:,6::-1], a, a[:,-1:-8:-1]))
c1 = np.hstack((a[:,6::-1], a))
c2 = np.hstack((a, a[:,-1:-8:-1]))
np.testing.assert_almost_equal(b, c)
np.testing.assert_almost_equal(b1, c)
np.testing.assert_almost_equal(b2, c1)
np.testing.assert_almost_equal(b3, c2)
b = pyyawt.wextend(1,'symw',a,7)
b1 = pyyawt.wextend(1,'symw',a,7,'b')
b2 = pyyawt.wextend(1,'symw',a,7,'l')
b3 = pyyawt.wextend(1,'symw',a,7,'r')
c = np.hstack((a[:,7:0:-1], a, a[:,-2:-9:-1]))
c1 = np.hstack((a[:,7:0:-1], a))
c2 = np.hstack((a, a[:,-2:-9:-1]))
np.testing.assert_almost_equal(b, c)
np.testing.assert_almost_equal(b1, c)
np.testing.assert_almost_equal(b2, c1)
np.testing.assert_almost_equal(b3, c2)
def test_vector(self):
a = np.random.randn(1, 51)
b = pyyawt.wextend(1, 'symh', a, 7)
b1 = pyyawt.wextend(1, 'symh', a, 7, 'b')
b2 = pyyawt.wextend(1, 'symh', a, 7, 'l')
b3 = pyyawt.wextend(1, 'symh', a, 7, 'r')
c = np.hstack((a[:, 6::-1], a, a[:, -1:-8:-1]))
c1 = np.hstack((a[:, 6::-1], a))
c2 = np.hstack((a, a[:, -1:-8:-1]))
np.testing.assert_almost_equal(b, c)
np.testing.assert_almost_equal(b1, c)
np.testing.assert_almost_equal(b2, c1)
np.testing.assert_almost_equal(b3, c2)
b = pyyawt.wextend(1, 'sym', a, 7)
b1 = pyyawt.wextend(1, 'sym', a, 7, 'b')
b2 = pyyawt.wextend(1, 'sym', a, 7, 'l')
b3 = pyyawt.wextend(1, 'sym', a, 7, 'r')
c = np.hstack((a[:, 6::-1], a, a[:, -1:-8:-1]))
c1 = np.hstack((a[:, 6::-1], a))
c2 = np.hstack((a, a[:, -1:-8:-1]))
np.testing.assert_almost_equal(b, c)
np.testing.assert_almost_equal(b1, c)
np.testing.assert_almost_equal(b2, c1)
np.testing.assert_almost_equal(b3, c2)
b = pyyawt.wextend(1, 'symw', a, 7)
b1 = pyyawt.wextend(1, 'symw', a, 7, 'b')
b2 = pyyawt.wextend(1, 'symw', a, 7, 'l')
b3 = pyyawt.wextend(1, 'symw', a, 7, 'r')
c = np.hstack((a[:, 7:0:-1], a, a[:, -2:-9:-1]))
c1 = np.hstack((a[:, 7:0:-1], a))
c2 = np.hstack((a, a[:, -2:-9:-1]))
np.testing.assert_almost_equal(b, c)
np.testing.assert_almost_equal(b1, c)
np.testing.assert_almost_equal(b2, c1)
np.testing.assert_almost_equal(b3, c2)
