diff options
Diffstat (limited to 'gr-blocks/python/blocks/qa_pack_k_bits.py')
| -rw-r--r-- | gr-blocks/python/blocks/qa_pack_k_bits.py | 29 |
1 files changed, 15 insertions, 14 deletions
diff --git a/gr-blocks/python/blocks/qa_pack_k_bits.py b/gr-blocks/python/blocks/qa_pack_k_bits.py index c4cfd6545e..5c5d077d44 100644 --- a/gr-blocks/python/blocks/qa_pack_k_bits.py +++ b/gr-blocks/python/blocks/qa_pack_k_bits.py @@ -14,6 +14,7 @@ import random from gnuradio import gr, gr_unittest, blocks import pmt + class test_pack(gr_unittest.TestCase): def setUp(self): @@ -56,27 +57,27 @@ class test_pack(gr_unittest.TestCase): def test_004(self): # Test tags propagation - #Tags on the incoming bits + # Tags on the incoming bits src_data = [1, 0, 1, 1, 0, 0, 0, 1] #src_tag_offsets = [1, 2, 3, 5, 6] src_tag_offsets = [1, 2, 3, 5, 6, 7] - #Ground Truth - expected_data= [2, 3, 0, 1] + # Ground Truth + expected_data = [2, 3, 0, 1] expected_tag_offsets = [0, 1, 1, 2, 3, 3] test_tags = list() tag_indexs = range(len(src_tag_offsets)) for src_tag in tag_indexs: test_tags.append( - gr.tag_utils.python_to_tag(( src_tag_offsets[src_tag], - pmt.intern('tag_byte'), - pmt.from_long(src_tag), - None - )) - ) - - src = blocks.vector_source_b(src_data, False,1, test_tags ) + gr.tag_utils.python_to_tag((src_tag_offsets[src_tag], + pmt.intern('tag_byte'), + pmt.from_long(src_tag), + None + )) + ) + + src = blocks.vector_source_b(src_data, False, 1, test_tags) op = blocks.pack_k_bits_bb(2) dst = blocks.vector_sink_b() self.tb.connect(src, op, dst) @@ -86,11 +87,11 @@ class test_pack(gr_unittest.TestCase): self.assertEqual(expected_data, dst.data()) # Check the tag values - self.assertEqual( list(tag_indexs), [ pmt.to_python(x.value) for x in dst.tags()]) + self.assertEqual(list(tag_indexs), [ + pmt.to_python(x.value) for x in dst.tags()]) # Check the tag offsets - self.assertEqual( expected_tag_offsets, [ x.offset for x in dst.tags()]) - + self.assertEqual(expected_tag_offsets, [x.offset for x in dst.tags()]) if __name__ == '__main__': |