git.gag.com Git - debian/gnuradio/blob - docs/howto-write-a-block/qa_howto_1.py

   1 #!/usr/bin/env python
   2
   3 from gnuradio import gr, gr_unittest
   4 import howto
   5
   6 class qa_howto (gr_unittest.TestCase):
   7
   8     def setUp (self):
   9         self.tb = gr.top_block ()
  10
  11     def tearDown (self):
  12         self.tb = None
  13
  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)
  22         self.tb.run ()
  23         result_data = dst.data ()
  24         self.assertFloatTuplesAlmostEqual (expected_result, result_data, 6)
  25
  26 if __name__ == '__main__':
  27     gr_unittest.main ()