3 # Copyright 2005,2006,2007,2010 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_cc(gr_unittest.TestCase):
28 self.tb = gr.top_block ()
34 src_data = (0+0j, 1000+1000j, 2000+2000j, 3000+3000j, 4000+4000j, 5000+5000j)
35 expected_result = src_data
36 src = gr.vector_source_c(src_data)
37 op = gr.single_pole_iir_filter_cc (1.0)
38 dst = gr.vector_sink_c()
39 self.tb.connect (src, op, dst)
41 result_data = dst.data()
42 self.assertComplexTuplesAlmostEqual (expected_result, result_data)
45 src_data = (complex(0,0), complex(1000,-1000), complex(2000,-2000), complex(3000,-3000), complex(4000,-4000), complex(5000,-5000))
46 expected_result = (complex(0,0), complex(125,-125), complex(359.375,-359.375), complex(689.453125,-689.453125), complex(1103.271484,-1103.271484), complex(1590.36255,-1590.36255))
47 src = gr.vector_source_c(src_data)
48 op = gr.single_pole_iir_filter_cc (0.125)
49 dst = gr.vector_sink_c()
50 self.tb.connect (src, op, dst)
52 result_data = dst.data()
53 self.assertComplexTuplesAlmostEqual (expected_result, result_data, 3)
57 src_data = (complex(0,0), complex(1000,-1000), complex(2000,-2000), complex(3000,-3000), complex(4000,-4000), complex(5000,-5000))
58 expected_result = (complex(0,0), complex(125,-125), complex(250,-250), complex(484.375,-484.375), complex(718.75,-718.75), complex(1048.828125,-1048.828125))
59 src = gr.vector_source_c(src_data)
60 s2p = gr.serial_to_parallel(gr.sizeof_gr_complex, block_size)
61 op = gr.single_pole_iir_filter_cc (0.125, block_size)
62 p2s = gr.parallel_to_serial(gr.sizeof_gr_complex, block_size)
63 dst = gr.vector_sink_c()
64 self.tb.connect (src, s2p, op, p2s, dst)
66 result_data = dst.data()
67 self.assertComplexTuplesAlmostEqual (expected_result, result_data, 3)
70 if __name__ == '__main__':
71 gr_unittest.run(test_single_pole_iir_cc, "test_single_pole_iir_cc.xml")