def test_coif(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 for N in np.arange(1,6): wname = "coif" + str(N) [cA,cD] = pyyawt.dwt(x1,wname) [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0) [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0) [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0)
def test_biorwavf(self): 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(len(wnames)): Lo_D,Hi_D,Lo_R,Hi_R = pyyawt.wfilters(wnames[N]) rf,df = pyyawt.biorwavf(wnames[N]) lo_d,hi_d,lo_r,hi_r = pyyawt.biorfilt(df,rf) np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D,Hi_D = pyyawt.wfilters(wnames[N],'d') Lo_R,Hi_R = pyyawt.wfilters(wnames[N],'r') np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D,Lo_R = pyyawt.wfilters(wnames[N],'l') Hi_D,Hi_R = pyyawt.wfilters(wnames[N],'h') np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R)
def test_coif(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 for N in np.arange(1, 6): wname = "coif" + str(N) [cA, cD] = pyyawt.dwt(x1, wname) [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0) [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0) [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0)
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) [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0) [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0) [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0)
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) [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0) [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0) [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0)
def test_haar(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 wname = 'haar' [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0) [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0) [cA, cD] = pyyawt.dwt(x1, wname) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 2 * np.size(cA) - np.size(Lo_R) + 2) r = pyyawt.idwt(cA, cD, wname) np.testing.assert_almost_equal(r, x0)
def test_haar(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 wname = 'haar' [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0) [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0) [cA,cD] = pyyawt.dwt(x1,wname) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,2*np.size(cA)-np.size(Lo_R)+2) r = pyyawt.idwt(cA,cD,wname) np.testing.assert_almost_equal(r, x0)
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_bior2(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 ST = pyyawt.dwtmode("status", "nodisp") 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(len(wnames)): wname = wnames[N] [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) [c, l] = pyyawt.wavedec(s1, 3, wname) s0 = pyyawt.waverec(c, l, wname) [cA1, cD1] = pyyawt.dwt(s1, wname) [cA2, cD2] = pyyawt.dwt(cA1, wname) [cA3, cD3] = pyyawt.dwt(cA2, wname) ca2 = pyyawt.idwt(cA3, cD3, wname, np.size(cA2)) ca1 = pyyawt.idwt(ca2, cD2, wname, np.size(cA1)) a0 = pyyawt.idwt(ca1, cD1, wname, np.size(s1)) np.testing.assert_almost_equal(a0, s0) Lo_D = np.random.randn(np.size(Lo_D)) Hi_D = np.random.randn(np.size(Lo_D)) ca2 = pyyawt.idwt(cA3, cD3, Lo_R, Hi_R, np.size(cA2)) ca1 = pyyawt.idwt(ca2, cD2, Lo_R, Hi_R, np.size(cA1)) a0 = pyyawt.idwt(ca1, cD1, Lo_R, Hi_R, np.size(s1)) s0 = pyyawt.waverec(c, l, Lo_R, Hi_R) np.testing.assert_almost_equal(a0, s0)
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_bior2(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 ST = pyyawt.dwtmode("status","nodisp") 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(len(wnames)): wname = wnames[N] [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) [c,l] = pyyawt.wavedec(s1,3,wname) s0 = pyyawt.waverec(c,l,wname) [cA1,cD1] = pyyawt.dwt(s1,wname) [cA2,cD2] = pyyawt.dwt(cA1,wname) [cA3,cD3] = pyyawt.dwt(cA2,wname) ca2 = pyyawt.idwt(cA3,cD3,wname,np.size(cA2)) ca1 = pyyawt.idwt(ca2,cD2,wname,np.size(cA1)) a0 = pyyawt.idwt(ca1,cD1,wname,np.size(s1)) np.testing.assert_almost_equal(a0, s0) Lo_D = np.random.randn(np.size(Lo_D)) Hi_D = np.random.randn(np.size(Lo_D)) ca2 = pyyawt.idwt(cA3,cD3,Lo_R,Hi_R,np.size(cA2)) ca1 = pyyawt.idwt(ca2,cD2,Lo_R,Hi_R,np.size(cA1)) a0 = pyyawt.idwt(ca1,cD1,Lo_R,Hi_R,np.size(s1)) s0 = pyyawt.waverec(c,l,Lo_R,Hi_R) np.testing.assert_almost_equal(a0, s0)
def test_biorwavf(self): 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(len(wnames)): Lo_D, Hi_D, Lo_R, Hi_R = pyyawt.wfilters(wnames[N]) rf, df = pyyawt.biorwavf(wnames[N]) lo_d, hi_d, lo_r, hi_r = pyyawt.biorfilt(df, rf) np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D, Hi_D = pyyawt.wfilters(wnames[N], "d") Lo_R, Hi_R = pyyawt.wfilters(wnames[N], "r") np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D, Lo_R = pyyawt.wfilters(wnames[N], "l") Hi_D, Hi_R = pyyawt.wfilters(wnames[N], "h") np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R)
def test_haar(self): Lo_D,Hi_D,Lo_R,Hi_R = pyyawt.wfilters('haar') w = pyyawt.dbwavf('db1') lo_d,hi_d,lo_r,hi_r = pyyawt.orthfilt(w) np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D,Hi_D = pyyawt.wfilters('haar','d') Lo_R,Hi_R = pyyawt.wfilters('haar','r') np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D,Lo_R = pyyawt.wfilters('haar','l') Hi_D,Hi_R = pyyawt.wfilters('haar','h') np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R)
def test_haar(self): Lo_D, Hi_D, Lo_R, Hi_R = pyyawt.wfilters("haar") w = pyyawt.dbwavf("db1") lo_d, hi_d, lo_r, hi_r = pyyawt.orthfilt(w) np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D, Hi_D = pyyawt.wfilters("haar", "d") Lo_R, Hi_R = pyyawt.wfilters("haar", "r") np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D, Lo_R = pyyawt.wfilters("haar", "l") Hi_D, Hi_R = pyyawt.wfilters("haar", "h") np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R)
def test_dbwavf(self): for N in np.arange(1, 37): Lo_D, Hi_D, Lo_R, Hi_R = pyyawt.wfilters("db" + str(N)) w = pyyawt.dbwavf("db" + str(N)) lo_d, hi_d, lo_r, hi_r = pyyawt.orthfilt(w) np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D, Hi_D = pyyawt.wfilters("db" + str(N), "d") Lo_R, Hi_R = pyyawt.wfilters("db" + str(N), "r") np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D, Lo_R = pyyawt.wfilters("db" + str(N), "l") Hi_D, Hi_R = pyyawt.wfilters("db" + str(N), "h") np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R)
def test_dbwavf(self): for N in np.arange(1,37): Lo_D,Hi_D,Lo_R,Hi_R = pyyawt.wfilters("db" + str(N)) w = pyyawt.dbwavf("db" + str(N)) lo_d,hi_d,lo_r,hi_r = pyyawt.orthfilt(w) np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D,Hi_D = pyyawt.wfilters("db" + str(N),'d') Lo_R,Hi_R = pyyawt.wfilters("db" + str(N),'r') np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R) Lo_D,Lo_R = pyyawt.wfilters("db" + str(N),'l') Hi_D,Hi_R = pyyawt.wfilters("db" + str(N),'h') np.testing.assert_almost_equal(lo_d, Lo_D) np.testing.assert_almost_equal(lo_r, Lo_R) np.testing.assert_almost_equal(hi_d, Hi_D) np.testing.assert_almost_equal(hi_r, Hi_R)
def test_type3(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 wname = "sym8" [cA, cD] = pyyawt.dwt(x1, wname) r = pyyawt.idwt(cA, cD, wname, 50) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 50) np.testing.assert_almost_equal(r, x0) [cA, cD] = pyyawt.dwt(x2, wname) r = pyyawt.idwt(cA, cD, wname, 50) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 50) np.testing.assert_almost_equal(r, x0) [cA, cD] = pyyawt.dwt(s1, wname) r = pyyawt.idwt(cA, cD, wname, 50) [Lo_D, Hi_D, Lo_R, Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 50) np.testing.assert_almost_equal(r, x0) Lo_R = np.random.randn(50) Hi_R = np.random.randn(50) a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 50) r = pyyawt.idwt(cA, cD, Lo_R.flatten(), Hi_R.flatten(), 50) np.testing.assert_almost_equal(r, x0)
def test_type3(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 wname = "sym8" [cA,cD] = pyyawt.dwt(x1,wname) r = pyyawt.idwt(cA,cD,wname,50) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,50) np.testing.assert_almost_equal(r, x0) [cA,cD] = pyyawt.dwt(x2,wname) r = pyyawt.idwt(cA,cD,wname,50) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,50) np.testing.assert_almost_equal(r, x0) [cA,cD] = pyyawt.dwt(s1,wname) r = pyyawt.idwt(cA,cD,wname,50) [Lo_D,Hi_D,Lo_R,Hi_R] = pyyawt.wfilters(wname) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,50) np.testing.assert_almost_equal(r, x0) Lo_R = np.random.randn(50) Hi_R = np.random.randn(50) a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,50) r = pyyawt.idwt(cA,cD,Lo_R.flatten(),Hi_R.flatten(),50) np.testing.assert_almost_equal(r, x0)
def test_type5(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 wname = "db7" dwtModes = ['symw', 'asymh', 'asymw', 'zpd', 'sp0', 'sp1', 'ppd','per'] [cA,cD] = pyyawt.dwt(x1,wname) [Lo_R,Hi_R] = pyyawt.wfilters(wname,'r') a0 = pyyawt.conv(pyyawt.dyadup(cA),Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD),Hi_R) x0 = pyyawt.wkeep(a0+d0,50) for N in np.arange(len(dwtModes)-1): r = pyyawt.idwt(cA,cD,wname,50,'mode',dwtModes[N]) np.testing.assert_almost_equal(r, x0) r = pyyawt.idwt(cA,cD,wname,50,'mode','per') x0 = pyyawt.wkeep(a0+d0,50) np.testing.assert_almost_equal(r, x0)
def test_type5(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 wname = "db7" dwtModes = [ 'symw', 'asymh', 'asymw', 'zpd', 'sp0', 'sp1', 'ppd', 'per' ] [cA, cD] = pyyawt.dwt(x1, wname) [Lo_R, Hi_R] = pyyawt.wfilters(wname, 'r') a0 = pyyawt.conv(pyyawt.dyadup(cA), Lo_R) d0 = pyyawt.conv(pyyawt.dyadup(cD), Hi_R) x0 = pyyawt.wkeep(a0 + d0, 50) for N in np.arange(len(dwtModes) - 1): r = pyyawt.idwt(cA, cD, wname, 50, 'mode', dwtModes[N]) np.testing.assert_almost_equal(r, x0) r = pyyawt.idwt(cA, cD, wname, 50, 'mode', 'per') x0 = pyyawt.wkeep(a0 + d0, 50) np.testing.assert_almost_equal(r, x0)
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_bior2(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 ST = pyyawt.dwtmode("status", "nodisp") 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(len(wnames)): wname = wnames[N] [cA1, cD1] = pyyawt.dwt(s1, wname) [cA2, cD2] = pyyawt.dwt(cA1, wname) [cA3, cD3] = pyyawt.dwt(cA2, wname) Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("asymh") [cA1, cD1] = pyyawt.dwt(s1, wname, 'mode', 'asymh') [cA2, cD2] = pyyawt.dwt(cA1, wname, 'mode', 'asymh') [cA3, cD3] = pyyawt.dwt(cA2, wname, 'mode', 'asymh') Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("symw") [cA1, cD1] = pyyawt.dwt(s1, wname, 'mode', 'symw') [cA2, cD2] = pyyawt.dwt(cA1, wname, 'mode', 'symw') [cA3, cD3] = pyyawt.dwt(cA2, wname, 'mode', 'symw') Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("asymw") [cA1, cD1] = pyyawt.dwt(s1, wname, 'mode', 'asymw') [cA2, cD2] = pyyawt.dwt(cA1, wname, 'mode', 'asymw') [cA3, cD3] = pyyawt.dwt(cA2, wname, 'mode', 'asymw') Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("zpd") [cA1, cD1] = pyyawt.dwt(s1, wname, 'mode', 'zpd') [cA2, cD2] = pyyawt.dwt(cA1, wname, 'mode', 'zpd') [cA3, cD3] = pyyawt.dwt(cA2, wname, 'mode', 'zpd') Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("sp0") [cA1, cD1] = pyyawt.dwt(s1, wname, 'mode', 'sp0') [cA2, cD2] = pyyawt.dwt(cA1, wname, 'mode', 'sp0') [cA3, cD3] = pyyawt.dwt(cA2, wname, 'mode', 'sp0') Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("sp1") [cA1, cD1] = pyyawt.dwt(s1, wname, 'mode', 'sp1') [cA2, cD2] = pyyawt.dwt(cA1, wname, 'mode', 'sp1') [cA3, cD3] = pyyawt.dwt(cA2, wname, 'mode', 'sp1') Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("ppd") [cA1, cD1] = pyyawt.dwt(s1, wname, 'mode', 'ppd') [cA2, cD2] = pyyawt.dwt(cA1, wname, 'mode', 'ppd') [cA3, cD3] = pyyawt.dwt(cA2, wname, 'mode', 'ppd') Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("per") [cA1, cD1] = pyyawt.dwt(s1, wname, 'mode', 'per') [cA2, cD2] = pyyawt.dwt(cA1, wname, 'mode', 'per') [cA3, cD3] = pyyawt.dwt(cA2, wname, 'mode', 'per') Lo_D, Hi_D = pyyawt.wfilters(wname, 'd') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [ np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1) ] c, l = pyyawt.wavedec(s1, 3, Lo_D, Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode(ST)
def test_bior2(self): s1 = self.s1 x1 = self.x1 x2 = self.x2 ST = pyyawt.dwtmode("status","nodisp") 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(len(wnames)): wname = wnames[N] [cA1,cD1] = pyyawt.dwt(s1,wname) [cA2,cD2] = pyyawt.dwt(cA1,wname) [cA3,cD3] = pyyawt.dwt(cA2,wname) Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("asymh") [cA1,cD1] = pyyawt.dwt(s1,wname,'mode','asymh') [cA2,cD2] = pyyawt.dwt(cA1,wname,'mode','asymh') [cA3,cD3] = pyyawt.dwt(cA2,wname,'mode','asymh') Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("symw") [cA1,cD1] = pyyawt.dwt(s1,wname,'mode','symw') [cA2,cD2] = pyyawt.dwt(cA1,wname,'mode','symw') [cA3,cD3] = pyyawt.dwt(cA2,wname,'mode','symw') Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("asymw") [cA1,cD1] = pyyawt.dwt(s1,wname,'mode','asymw') [cA2,cD2] = pyyawt.dwt(cA1,wname,'mode','asymw') [cA3,cD3] = pyyawt.dwt(cA2,wname,'mode','asymw') Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("zpd") [cA1,cD1] = pyyawt.dwt(s1,wname,'mode','zpd') [cA2,cD2] = pyyawt.dwt(cA1,wname,'mode','zpd') [cA3,cD3] = pyyawt.dwt(cA2,wname,'mode','zpd') Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("sp0") [cA1,cD1] = pyyawt.dwt(s1,wname,'mode','sp0') [cA2,cD2] = pyyawt.dwt(cA1,wname,'mode','sp0') [cA3,cD3] = pyyawt.dwt(cA2,wname,'mode','sp0') Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("sp1") [cA1,cD1] = pyyawt.dwt(s1,wname,'mode','sp1') [cA2,cD2] = pyyawt.dwt(cA1,wname,'mode','sp1') [cA3,cD3] = pyyawt.dwt(cA2,wname,'mode','sp1') Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("ppd") [cA1,cD1] = pyyawt.dwt(s1,wname,'mode','ppd') [cA2,cD2] = pyyawt.dwt(cA1,wname,'mode','ppd') [cA3,cD3] = pyyawt.dwt(cA2,wname,'mode','ppd') Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode("per") [cA1,cD1] = pyyawt.dwt(s1,wname,'mode','per') [cA2,cD2] = pyyawt.dwt(cA1,wname,'mode','per') [cA3,cD3] = pyyawt.dwt(cA2,wname,'mode','per') Lo_D,Hi_D = pyyawt.wfilters(wname,'d') c0 = np.concatenate([cA3, cD3, cD2, cD1]) l0 = [np.size(cA3), np.size(cD3), np.size(cD2), np.size(cD1), np.size(s1)] c, l = pyyawt.wavedec(s1,3,Lo_D,Hi_D) np.testing.assert_almost_equal(c, c0) np.testing.assert_almost_equal(l, l0) pyyawt.dwtmode(ST)