def test_firpm_1(self): x = [i for i in range(len(self.f1))] ipf = ip.interp1d(x, self.f1) f1_new = ipf(np.linspace(x[0], x[-1], 2 * len(x))) FIR = FIRDesign.firpm(self.n1, self.f1, self.a1) fir = signal.remez(self.n1 + 1, f1_new, self.a1, fs=2) self.assertTrue(np.all(FIR[0] == fir))
def test_firpm_2(self): f2_2 = self.f2 / (self.fs / 2) f2_2[0] = 0 f2_2[-1] = 1 x = [i for i in range(len(f2_2))] ipf = ip.interp1d(x, f2_2) f2_new = ipf(np.linspace(x[0], x[-1], 2 * len(x))) x_w = [i for i in range(len(self.w))] ipw = ip.interp1d(x_w, self.w) w_new = ipw(np.linspace(x_w[0], x_w[-1], 2 * len(x_w))) FIR = FIRDesign.firpm(self.n2, f2_2, self.a2, self.w) fir = signal.remez(self.n2 + 1, f2_new, self.a2, w_new, fs=2) self.assertTrue(np.all(FIR[0] == fir))