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))
