3 from gnuradio import gr, gr_unittest
6 class qa_howto (gr_unittest.TestCase):
9 self.tb = gr.top_block ()
14 def test_001_square_ff (self):
15 src_data = (-3, 4, -5.5, 2, 3)
16 expected_result = (9, 16, 30.25, 4, 9)
17 src = gr.vector_source_f (src_data)
18 sqr = howto.square_ff ()
19 dst = gr.vector_sink_f ()
20 self.tb.connect (src, sqr)
21 self.tb.connect (sqr, dst)
23 result_data = dst.data ()
24 self.assertFloatTuplesAlmostEqual (expected_result, result_data, 6)
26 if __name__ == '__main__':