def test_002_work_function_packed(self): block_size = 256 num_info_bits = block_size // 2 data, ref, polar_encoder = self.get_test_data(block_size, num_info_bits, 1) 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() res = np.array(snk.data()).astype(dtype=int) self.assertTupleEqual(tuple(res), tuple(ref))
def test_004_big_input(self): 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) 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() res = np.array(snk.data()).astype(dtype=int) self.assertTupleEqual(tuple(res), tuple(ref))