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_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))
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))