diff options
| author | Tom Rondeau <trondeau@vt.edu> | 2013-03-16 10:30:16 -0400 |
|---|---|---|
| committer | Tom Rondeau <trondeau@vt.edu> | 2013-03-16 10:30:16 -0400 |
| commit | 7774331476979a530288eb8217d19f8f7af64ed1 (patch) | |
| tree | 4acb7c1df73a0d8d61633eda3b1108217f54f03b /gr-digital/python/qa_header_payload_demux.py | |
| parent | 34216336205ba3edc0bc308a8da9ad388f3d0774 (diff) | |
| parent | 4a030cb4866c651ebe5c53a6a429cf2b5e0c11bd (diff) | |
Merge branch 'master' of gnuradio.org:gnuradio
Diffstat (limited to 'gr-digital/python/qa_header_payload_demux.py')
| -rwxr-xr-x | gr-digital/python/qa_header_payload_demux.py | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/gr-digital/python/qa_header_payload_demux.py b/gr-digital/python/qa_header_payload_demux.py new file mode 100755 index 0000000000..15b3eadcf8 --- /dev/null +++ b/gr-digital/python/qa_header_payload_demux.py @@ -0,0 +1,78 @@ +#!/usr/bin/env python +# Copyright 2012 Free Software Foundation, Inc. +# +# This file is part of GNU Radio +# +# GNU Radio is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 3, or (at your option) +# any later version. +# +# GNU Radio is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with GNU Radio; see the file COPYING. If not, write to +# the Free Software Foundation, Inc., 51 Franklin Street, +# Boston, MA 02110-1301, USA. +# + +from gnuradio import gr, gr_unittest +try: import pmt +except: from gruel import pmt +import digital_swig as digital +import time + +class qa_header_payload_demux (gr_unittest.TestCase): + + def setUp (self): + self.tb = gr.top_block () + + def tearDown (self): + self.tb = None + + def test_001_t (self): + """ Simplest possible test: put in zeros, then header, + then payload, trigger signal, try to demux. + The return signal from the header parser is faked via _post() + """ + n_zeros = 100 + header = (1, 2, 3) + payload = tuple(range(17)) + data_signal = (0,) * n_zeros + header + payload + trigger_signal = [0,] * len(data_signal) + trigger_signal[n_zeros] = 1 + + data_src = gr.vector_source_f(data_signal, False) + trigger_src = gr.vector_source_b(trigger_signal, False) + hpd = digital.header_payload_demux( + len(header), 1, 0, "frame_len", "detect", False, gr.sizeof_float + ) + self.assertEqual(pmt.pmt_length(hpd.message_ports_in()), 1) + header_sink = gr.vector_sink_f() + payload_sink = gr.vector_sink_f() + + self.tb.connect(data_src, (hpd, 0)) + self.tb.connect(trigger_src, (hpd, 1)) + self.tb.connect((hpd, 0), header_sink) + self.tb.connect((hpd, 1), payload_sink) + self.tb.start() + time.sleep(.2) # Need this, otherwise, the next message is ignored + hpd.to_basic_block()._post( + pmt.pmt_intern('header_data'), + pmt.pmt_from_long(len(payload)) + ) + while len(payload_sink.data()) < len(payload): + time.sleep(.2) + self.tb.stop() + self.tb.wait() + + self.assertEqual(header_sink.data(), header) + self.assertEqual(payload_sink.data(), payload) + + +if __name__ == '__main__': + gr_unittest.run(qa_header_payload_demux, "qa_header_payload_demux.xml") + |