diff options
Diffstat (limited to 'gr-fec/python/fec/qa_polar_encoder.py')
| -rw-r--r-- | gr-fec/python/fec/qa_polar_encoder.py | 23 |
1 files changed, 5 insertions, 18 deletions
diff --git a/gr-fec/python/fec/qa_polar_encoder.py b/gr-fec/python/fec/qa_polar_encoder.py index 03629354f0..4af193e2d4 100644 --- a/gr-fec/python/fec/qa_polar_encoder.py +++ b/gr-fec/python/fec/qa_polar_encoder.py @@ -22,14 +22,11 @@ from gnuradio import gr, gr_unittest, blocks import fec_swig as fec -from _qa_helper import _qa_helper import numpy as np from extended_encoder import extended_encoder -from extended_decoder import extended_decoder from polar.encoder import PolarEncoder from polar.helper_functions import get_frozen_bit_positions -from polar.helper_functions import bit_reverse_vector class test_polar_encoder(gr_unittest.TestCase): @@ -61,21 +58,17 @@ class test_polar_encoder(gr_unittest.TestCase): frozen_bit_values = np.array([0] * num_frozen_bits,) python_encoder = PolarEncoder(block_size, num_info_bits, frozen_bit_positions, frozen_bit_values) - is_packed = False + is_packed = True polar_encoder = fec.polar_encoder.make(block_size, num_info_bits, frozen_bit_positions, frozen_bit_values, is_packed) data = np.ones(num_info_bits, dtype=int) src = blocks.vector_source_b(data, False) - packer = blocks.pack_k_bits_bb(8) enc_block = extended_encoder(polar_encoder, None, '11') - unpacker = blocks.unpack_k_bits_bb(8) snk = blocks.vector_sink_b(1) - if is_packed: - self.tb.connect(src, packer, enc_block, unpacker, snk) - else: - self.tb.connect(src, enc_block, snk) + self.tb.connect(src, enc_block, snk) self.tb.run() + print(self.tb.edge_list()) res = np.array(snk.data()).astype(dtype=int) penc = python_encoder.encode(data) @@ -85,7 +78,7 @@ class test_polar_encoder(gr_unittest.TestCase): self.assertTupleEqual(tuple(res), tuple(penc)) def test_003_big_input(self): - is_packed = False + is_packed = True num_blocks = 30 block_size = 256 num_info_bits = 128 @@ -106,19 +99,13 @@ class test_polar_encoder(gr_unittest.TestCase): src = blocks.vector_source_b(data, False) - packer = blocks.pack_k_bits_bb(8) enc_block = extended_encoder(polar_encoder, None, '11') - unpacker = blocks.unpack_k_bits_bb(8) snk = blocks.vector_sink_b(1) - if is_packed: - self.tb.connect(src, packer, enc_block, unpacker, snk) - else: - self.tb.connect(src, enc_block, snk) + self.tb.connect(src, enc_block, snk) self.tb.run() res = np.array(snk.data()).astype(dtype=int) - # penc = python_encoder.encode(data) print(res) print(ref) |