git.gag.com Git - debian/gnuradio/blob - gnuradio-core/src/python/gnuradio/gr/qa_cma_equalizer.py

   1 #!/usr/bin/env python
   2
   3 from gnuradio import gr, gr_unittest
   4
   5 class test_cma_equalizer_fir(gr_unittest.TestCase):
   6
   7     def setUp(self):
   8         self.fg = gr.flow_graph()
   9
  10     def tearDown(self):
  11         self.fg = None
  12
  13     def transform(self, src_data):
  14         SRC = gr.vector_source_c(src_data, False)
  15         EQU = gr.cma_equalizer_cc(4, 1.0, .001)
  16         DST = gr.vector_sink_c()
  17         self.fg.connect(SRC, EQU, DST)
  18         self.fg.run()
  19         return DST.data()
  20
  21     def test_001_identity(self):
  22         # Constant modulus signal so no adjustments
  23         src_data      = (1+0j, 0+1j, -1+0j, 0-1j)*1000
  24         expected_data = src_data
  25         result = self.transform(src_data)
  26         self.assertComplexTuplesAlmostEqual(expected_data, result)
  27
  28 if __name__ == "__main__":
  29     gr_unittest.main()