diff options
Diffstat (limited to 'gr-fec/python/fec/qa_polar_encoder.py')
| -rw-r--r-- | gr-fec/python/fec/qa_polar_encoder.py | 73 |
1 files changed, 42 insertions, 31 deletions
diff --git a/gr-fec/python/fec/qa_polar_encoder.py b/gr-fec/python/fec/qa_polar_encoder.py index b4f26e4017..90190cd719 100644 --- a/gr-fec/python/fec/qa_polar_encoder.py +++ b/gr-fec/python/fec/qa_polar_encoder.py @@ -28,6 +28,10 @@ from extended_encoder import extended_encoder from polar.encoder import PolarEncoder import polar.channel_construction as cc +# import os +# print('PID:', os.getpid()) +# raw_input('tell me smth') + class test_polar_encoder(gr_unittest.TestCase): @@ -50,54 +54,45 @@ class test_polar_encoder(gr_unittest.TestCase): self.assertFloatTuplesAlmostEqual((float(num_info_bits) / block_size, ), (polar_encoder.rate(), )) self.assertFalse(polar_encoder.set_frame_size(10)) - def test_002_work_function(self): - block_size = 256 - num_info_bits = 128 - num_frozen_bits = block_size - num_info_bits - frozen_bit_positions = cc.frozen_bit_positions(block_size, num_info_bits, 0.0) - frozen_bit_values = np.array([0] * num_frozen_bits,) - python_encoder = PolarEncoder(block_size, num_info_bits, frozen_bit_positions, frozen_bit_values) - + def test_002_work_function_packed(self): is_packed = True - polar_encoder = fec.polar_encoder.make(block_size, num_info_bits, frozen_bit_positions, frozen_bit_values, is_packed) + block_size = 256 + num_info_bits = block_size // 2 - data = np.ones(num_info_bits, dtype=int) + data, ref, polar_encoder = self.get_test_data(block_size, num_info_bits, 1, is_packed) src = blocks.vector_source_b(data, False) enc_block = extended_encoder(polar_encoder, None, '11') snk = blocks.vector_sink_b(1) 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) - - print(res) - print(penc) - self.assertTupleEqual(tuple(res), tuple(penc)) + self.assertTupleEqual(tuple(res), tuple(ref)) - def test_003_big_input(self): - is_packed = True - num_blocks = 30 + def test_003_work_function_unpacked(self): + is_packed = False block_size = 256 - num_info_bits = 128 - num_frozen_bits = block_size - num_info_bits - frozen_bit_positions = cc.frozen_bit_positions(block_size, num_info_bits, 0.0) - frozen_bit_values = np.array([0] * num_frozen_bits,) - python_encoder = PolarEncoder(block_size, num_info_bits, frozen_bit_positions, frozen_bit_values) + num_info_bits = block_size // 2 - polar_encoder = fec.polar_encoder.make(block_size, num_info_bits, frozen_bit_positions, frozen_bit_values, is_packed) + data, ref, polar_encoder = self.get_test_data(block_size, num_info_bits, 1, is_packed) + src = blocks.vector_source_b(data, False) + enc_block = extended_encoder(polar_encoder, None, '11') + snk = blocks.vector_sink_b(1) - data = np.array([], dtype=int) - ref = np.array([], dtype=int) + self.tb.connect(src, enc_block, snk) + self.tb.run() - for i in range(num_blocks): - d = np.random.randint(2, size=num_info_bits) - data = np.append(data, d) - ref = np.append(ref, python_encoder.encode(d)) + res = np.array(snk.data()).astype(dtype=int) + self.assertTupleEqual(tuple(res), tuple(ref)) + def test_004_big_input(self): + is_packed = False + num_blocks = 30 + block_size = 1024 + num_info_bits = block_size // 8 + data, ref, polar_encoder = self.get_test_data(block_size, num_info_bits, num_blocks, is_packed) src = blocks.vector_source_b(data, False) enc_block = extended_encoder(polar_encoder, None, '11') snk = blocks.vector_sink_b(1) @@ -111,6 +106,22 @@ class test_polar_encoder(gr_unittest.TestCase): print(ref) self.assertTupleEqual(tuple(res), tuple(ref)) + def get_test_data(self, block_size, num_info_bits, num_blocks, is_packed): + num_frozen_bits = block_size - num_info_bits + frozen_bit_positions = cc.frozen_bit_positions(block_size, num_info_bits, 0.0) + frozen_bit_values = np.array([0] * num_frozen_bits,) + python_encoder = PolarEncoder(block_size, num_info_bits, frozen_bit_positions, frozen_bit_values) + + data = np.array([], dtype=int) + ref = np.array([], dtype=int) + for i in range(num_blocks): + d = np.random.randint(2, size=num_info_bits) + data = np.append(data, d) + ref = np.append(ref, python_encoder.encode(d)) + polar_encoder = fec.polar_encoder.make(block_size, num_info_bits, frozen_bit_positions, frozen_bit_values, is_packed) + return data, ref, polar_encoder + + |