def test_iirnotch_2(self):
# Test case for Exception 1
with self.assertRaises(ValueError):
IIRDesign.iirnotch(4, self.bw)
def test_iirnotch_3(self):
# Test case for Exception 2
with self.assertRaises(ValueError):
IIRDesign.iirnotch(self.w0, 40)
def test_iirnotch_1(self):
# Test case
IIR = IIRDesign.iirnotch(self.w0, self.bw)
iir = signal.iirnotch(self.w0, self.w0 / self.bw, fs=2)
self.assertTrue(np.all(IIR[0] == iir[0]) and np.all(IIR[1] == iir[1]))
