Python测试驱动编写

通过gr_modtool会自动生成测试驱动程序的模板，只需要修改模板即可，模板如下：

from gnuradio import gr, gr_unittest
from gnuradio import blocks
import double_diff_swig as double_diff

class qa_translate (gr_unittest.TestCase):

    def setUp (self):
        self.tb = gr.top_block ()

    def tearDown (self):
        self.tb = None

    def test_001_t (self):
        # set up fg
        self.tb.run ()
        # check data


if __name__ == '__main__':
    gr_unittest.run(qa_translate, "qa_translate.xml")

我们需要做的就是补充test_001_t()函数：构建流图、给出测试数据以及预期结果、运行流图、将实际结果与预期结果I进行比较。修改好的程序如下：

from gnuradio import gr, gr_unittest
from gnuradio import blocks
import double_diff

class qa_translate (gr_unittest.TestCase):

    def setUp (self):
        self.tb = gr.top_block ()

    def tearDown (self):
        self.tb = None

    def test_001_t (self):
	src_data=[1,-1,-1,1,1,-1]
	expected_ampl=(1,1,1,1,1,1)
	expected_phase=(0,1,1,0,0,1)

	src=blocks.vector_source_c(src_data)
	op=double_diff.translate()
	dst0=blocks.vector_sink_f()
	dst1=blocks.vector_sink_f()

	self.tb.connect(src,op)
	self.tb.connect((op,0),dst0)
	self.tb.connect((op,1),dst1)
        # set up fg
        self.tb.run ()
        # check data
	result_ampl=dst0.data()
	result_phase=dst1.data()
	self.assertEqual(expected_ampl,result_ampl)
	self.assertEqual(expected_phase,result_phase)


if __name__ == '__main__':
    gr_unittest.run(qa_translate, "qa_translate.xml")