def test_parse_filter(self): fir = numpy.arange(10) assert filter_design.parse_filter(fir) == ('ba', (fir, [1.])) zpk = ([1, 2, 3], [4, 5, 6], 1.) parsed = filter_design.parse_filter(zpk) assert parsed[0] == 'zpk' utils.assert_zpk_equal(parsed[1], zpk)
def test_parse_filter(self): fir = numpy.arange(10) assert filter_design.parse_filter(fir) == ('ba', (fir, [1.])) zpk = ([1, 2, 3], [4, 5, 6], 1.) parsed = filter_design.parse_filter(zpk) assert parsed[0] == 'zpk' utils.assert_zpk_equal(parsed[1], zpk)
def test_highpass(self): iir = gwpy_signal.highpass(100, 1024) utils.assert_zpk_equal(iir, HIGHPASS_IIR_100HZ) fir = gwpy_signal.highpass(100, 1024, type='fir') print(fir) print(HIGHPASS_FIR_100HZ) print(fir - HIGHPASS_FIR_100HZ) utils.assert_allclose(fir, HIGHPASS_FIR_100HZ)
def test_notch_design(self): """Test :func:`gwpy.signal.filter_design.notch` """ # test simple notch zpk = filter_design.notch(60, 16384) utils.assert_zpk_equal(zpk, NOTCH_60HZ) # test Quantities zpk2 = filter_design.notch(60 * ONE_HZ, 16384 * ONE_HZ) utils.assert_zpk_equal(zpk, zpk2) # test FIR notch doesn't work with pytest.raises(NotImplementedError): filter_design.notch(60, 16384, type='fir')
def test_notch_design(self): """Test :func:`gwpy.signal.filter_design.notch` """ # test simple notch zpk = gwpy_signal.notch(60, 16384) utils.assert_zpk_equal(zpk, NOTCH_60HZ) # test Quantities zpk2 = gwpy_signal.notch(60 * ONE_HZ, 16384 * ONE_HZ) utils.assert_zpk_equal(zpk, zpk2) # test FIR notch doesn't work with pytest.raises(NotImplementedError): gwpy_signal.notch(60, 16384, type='fir')
def test_concatenate_zpks(self): zpk1 = ([1, 2, 3], [4, 5, 6], 1.) zpk2 = ([1, 2, 3, 4], [5, 6, 7, 8], 100) utils.assert_zpk_equal( filter_design.concatenate_zpks(zpk1, zpk2), ([1, 2, 3, 1, 2, 3, 4], [4, 5, 6, 5, 6, 7, 8], 100))
def test_bandpass(self): iir = filter_design.bandpass(100, 200, 1024) utils.assert_zpk_equal(iir, BANDPASS_IIR_100HZ_200HZ) fir = filter_design.bandpass(100, 200, 1024, type='fir') utils.assert_allclose(fir, BANDPASS_FIR_100HZ_200HZ)
def test_highpass(self): iir = filter_design.highpass(100, 1024) utils.assert_zpk_equal(iir, HIGHPASS_IIR_100HZ) fir = filter_design.highpass(100, 1024, type='fir') utils.assert_allclose(fir, HIGHPASS_FIR_100HZ)
def test_concatenate_zpks(self): zpk1 = ([1, 2, 3], [4, 5, 6], 1.) zpk2 = ([1, 2, 3, 4], [5, 6, 7, 8], 100) utils.assert_zpk_equal(gwpy_signal.concatenate_zpks( zpk1, zpk2), ([1, 2, 3, 1, 2, 3, 4], [4, 5, 6, 5, 6, 7, 8], 100))
def test_bandpass(self): iir = gwpy_signal.bandpass(100, 200, 1024) utils.assert_zpk_equal(iir, BANDPASS_IIR_100HZ_200HZ) fir = gwpy_signal.bandpass(100, 200, 1024, type='fir') utils.assert_allclose(fir, BANDPASS_FIR_100HZ_200HZ)
def test_lowpass(self): iir = gwpy_signal.lowpass(100, 1024) utils.assert_zpk_equal(iir, LOWPASS_IIR_100HZ) fir = gwpy_signal.lowpass(100, 1024, type='fir') utils.assert_allclose(fir, LOWPASS_FIR_100HZ)
def test_highpass(self): iir = filter_design.highpass(100, 1024) utils.assert_zpk_equal(iir, HIGHPASS_IIR_100HZ) fir = filter_design.highpass(100, 1024, type='fir') utils.assert_allclose(fir, HIGHPASS_FIR_100HZ)