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 | 23 |
1 files changed, 12 insertions, 11 deletions
diff --git a/gr-blocks/python/blocks/qa_pack_k_bits.py b/gr-blocks/python/blocks/qa_pack_k_bits.py index 435325cd40..b38f9c519d 100644 --- a/gr-blocks/python/blocks/qa_pack_k_bits.py +++ b/gr-blocks/python/blocks/qa_pack_k_bits.py @@ -13,6 +13,7 @@ import random from gnuradio import gr, gr_unittest, blocks + class test_pack(gr_unittest.TestCase): def setUp(self): @@ -23,9 +24,9 @@ class test_pack(gr_unittest.TestCase): self.tb = None def test_001(self): - src_data = [1,0,1,1,0,1,1,0] - expected_results = [1,0,1,1,0,1,1,0] - src = blocks.vector_source_b(src_data,False) + src_data = [1, 0, 1, 1, 0, 1, 1, 0] + expected_results = [1, 0, 1, 1, 0, 1, 1, 0] + src = blocks.vector_source_b(src_data, False) op = blocks.pack_k_bits_bb(1) dst = blocks.vector_sink_b() self.tb.connect(src, op, dst) @@ -33,9 +34,9 @@ class test_pack(gr_unittest.TestCase): self.assertEqual(expected_results, dst.data()) def test_002(self): - src_data = [1,0,1,1,0,0,0,1] - expected_results = [ 2, 3, 0, 1] - src = blocks.vector_source_b(src_data,False) + src_data = [1, 0, 1, 1, 0, 0, 0, 1] + expected_results = [2, 3, 0, 1] + src = blocks.vector_source_b(src_data, False) op = blocks.pack_k_bits_bb(2) dst = blocks.vector_sink_b() self.tb.connect(src, op, dst) @@ -44,15 +45,15 @@ class test_pack(gr_unittest.TestCase): self.assertEqual(expected_results, dst.data()) def test_003(self): - src_data = expected_results = [random.randint(0,3) for x in range(10)]; - src = blocks.vector_source_b( src_data ) + src_data = expected_results = [random.randint(0, 3) for x in range(10)] + src = blocks.vector_source_b(src_data) pack = blocks.pack_k_bits_bb(2) unpack = blocks.unpack_k_bits_bb(2) snk = blocks.vector_sink_b() - self.tb.connect(src,unpack,pack,snk) + self.tb.connect(src, unpack, pack, snk) self.tb.run() self.assertEqual(list(expected_results), list(snk.data())) -if __name__ == '__main__': - gr_unittest.run(test_pack) +if __name__ == '__main__': + gr_unittest.run(test_pack) |