diff options
Diffstat (limited to 'gr-howto-write-a-block/python/qa_howto.py')
| -rw-r--r--[-rwxr-xr-x] | gr-howto-write-a-block/python/qa_howto.py | 50 |
1 files changed, 25 insertions, 25 deletions
diff --git a/gr-howto-write-a-block/python/qa_howto.py b/gr-howto-write-a-block/python/qa_howto.py index 5e62fc890d..203b06922b 100755..100644 --- a/gr-howto-write-a-block/python/qa_howto.py +++ b/gr-howto-write-a-block/python/qa_howto.py @@ -1,6 +1,6 @@ #!/usr/bin/env python # -# Copyright 2004,2007 Free Software Foundation, Inc. +# Copyright 2004,2007,2012 Free Software Foundation, Inc. # # This file is part of GNU Radio # @@ -21,40 +21,40 @@ # from gnuradio import gr, gr_unittest -import howto_swig +import howto_swig as howto from square3_ff import square3_ff -class qa_howto (gr_unittest.TestCase): +class qa_howto(gr_unittest.TestCase): - def setUp (self): - self.tb = gr.top_block () + def setUp(self): + self.tb = gr.top_block() - def tearDown (self): + def tearDown(self): self.tb = None - def test_001_square_ff (self): + def test_001_square_ff(self): src_data = (-3, 4, -5.5, 2, 3) expected_result = (9, 16, 30.25, 4, 9) - src = gr.vector_source_f (src_data) - sqr = howto_swig.square_ff () - dst = gr.vector_sink_f () - self.tb.connect (src, sqr) - self.tb.connect (sqr, dst) - self.tb.run () - result_data = dst.data () - self.assertFloatTuplesAlmostEqual (expected_result, result_data, 6) + src = gr.vector_source_f(src_data) + sqr = howto.square_ff() + dst = gr.vector_sink_f() + self.tb.connect(src, sqr) + self.tb.connect(sqr, dst) + self.tb.run() + result_data = dst.data() + self.assertFloatTuplesAlmostEqual(expected_result, result_data, 6) - def test_002_square2_ff (self): + def test_002_square2_ff(self): src_data = (-3, 4, -5.5, 2, 3) expected_result = (9, 16, 30.25, 4, 9) - src = gr.vector_source_f (src_data) - sqr = howto_swig.square2_ff () - dst = gr.vector_sink_f () - self.tb.connect (src, sqr) - self.tb.connect (sqr, dst) - self.tb.run () - result_data = dst.data () - self.assertFloatTuplesAlmostEqual (expected_result, result_data, 6) + src = gr.vector_source_f(src_data) + sqr = howto.square2_ff() + dst = gr.vector_sink_f() + self.tb.connect(src, sqr) + self.tb.connect(sqr, dst) + self.tb.run() + result_data = dst.data() + self.assertFloatTuplesAlmostEqual(expected_result, result_data, 6) def test_003_square3_ff (self): src_data = (-3, 4, -5.5, 2, 3) @@ -69,4 +69,4 @@ class qa_howto (gr_unittest.TestCase): self.assertFloatTuplesAlmostEqual (expected_result, result_data, 6) if __name__ == '__main__': - gr_unittest.main () + gr_unittest.run(qa_howto, "qa_howto.xml") |