3 # Copyright 2005,2007 Free Software Foundation, Inc.
5 # This file is part of GNU Radio
7 # GNU Radio is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 3, or (at your option)
12 # GNU Radio is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with GNU Radio; see the file COPYING. If not, write to
19 # the Free Software Foundation, Inc., 51 Franklin Street,
20 # Boston, MA 02110-1301, USA.
23 from gnuradio import gr, gr_unittest
25 class test_single_pole_iir(gr_unittest.TestCase):
28 self.tb = gr.top_block ()
34 src_data = (0, 1000, 2000, 3000, 4000, 5000)
35 expected_result = src_data
36 src = gr.vector_source_f(src_data)
37 op = gr.single_pole_iir_filter_ff (1.0)
38 dst = gr.vector_sink_f()
39 self.tb.connect (src, op, dst)
41 result_data = dst.data()
42 self.assertFloatTuplesAlmostEqual (expected_result, result_data)
45 src_data = (0, 1000, 2000, 3000, 4000, 5000)
46 expected_result = (0, 125, 359.375, 689.453125, 1103.271484, 1590.36255)
47 src = gr.vector_source_f(src_data)
48 op = gr.single_pole_iir_filter_ff (0.125)
49 dst = gr.vector_sink_f()
50 self.tb.connect (src, op, dst)
52 result_data = dst.data()
53 self.assertFloatTuplesAlmostEqual (expected_result, result_data, 3)
57 src_data = (0, 1000, 2000, 3000, 4000, 5000)
58 expected_result = (0, 125, 250, 484.375, 718.75, 1048.828125)
59 src = gr.vector_source_f(src_data)
60 s2p = gr.serial_to_parallel(gr.sizeof_float, block_size)
61 op = gr.single_pole_iir_filter_ff (0.125, block_size)
62 p2s = gr.parallel_to_serial(gr.sizeof_float, block_size)
63 dst = gr.vector_sink_f()
64 self.tb.connect (src, s2p, op, p2s, dst)
66 result_data = dst.data()
67 self.assertFloatTuplesAlmostEqual (expected_result, result_data, 3)
70 if __name__ == '__main__':