diff options
| author | mormj <mormjb@gmail.com> | 2020-10-30 10:59:50 -0400 |
|---|---|---|
| committer | Marcus Müller <marcus@hostalia.de> | 2020-10-30 17:52:53 +0100 |
| commit | 7a0948ba85758fba1cc3858ef99bfa600dcc7416 (patch) | |
| tree | 610d7f9d773a193562def6df2d4b50f1bb3b3f86 /gr-fec/python/fec/qa_polar_encoder_systematic.py | |
| parent | 12192ee7d58de95ddca35a3e93bfc172bdb5c820 (diff) | |
qa: run autopep8 formatting on qa python files
find ./ -iname qa*.py | xargs autopep8 --in-place -a -a
mostly formats whitespace and gets rid of trailing semicolons
Diffstat (limited to 'gr-fec/python/fec/qa_polar_encoder_systematic.py')
| -rw-r--r-- | gr-fec/python/fec/qa_polar_encoder_systematic.py | 27 |
1 files changed, 19 insertions, 8 deletions
diff --git a/gr-fec/python/fec/qa_polar_encoder_systematic.py b/gr-fec/python/fec/qa_polar_encoder_systematic.py index 4118567216..25a8302b6b 100644 --- a/gr-fec/python/fec/qa_polar_encoder_systematic.py +++ b/gr-fec/python/fec/qa_polar_encoder_systematic.py @@ -32,20 +32,24 @@ class test_polar_encoder_systematic(gr_unittest.TestCase): def test_001_setup(self): block_size = 16 num_info_bits = 8 - frozen_bit_positions = cc.frozen_bit_positions(block_size, num_info_bits, 0.0) + frozen_bit_positions = cc.frozen_bit_positions( + block_size, num_info_bits, 0.0) - polar_encoder = fec.polar_encoder_systematic.make(block_size, num_info_bits, frozen_bit_positions) + polar_encoder = fec.polar_encoder_systematic.make( + block_size, num_info_bits, frozen_bit_positions) self.assertEqual(block_size, polar_encoder.get_output_size()) self.assertEqual(num_info_bits, polar_encoder.get_input_size()) - self.assertFloatTuplesAlmostEqual((float(num_info_bits) / block_size, ), (polar_encoder.rate(), )) + self.assertFloatTuplesAlmostEqual( + (float(num_info_bits) / block_size, ), (polar_encoder.rate(), )) self.assertFalse(polar_encoder.set_frame_size(10)) 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) + 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) @@ -61,7 +65,8 @@ class test_polar_encoder_systematic(gr_unittest.TestCase): block_size = 1024 num_info_bits = block_size // 8 - data, ref, polar_encoder = self.get_test_data(block_size, num_info_bits, num_blocks) + 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) @@ -75,9 +80,14 @@ class test_polar_encoder_systematic(gr_unittest.TestCase): def get_test_data(self, block_size, num_info_bits, num_blocks): # helper function to set up test data and together with encoder object. num_frozen_bits = block_size - num_info_bits - frozen_bit_positions = cc.frozen_bit_positions(block_size, num_info_bits, 0.0) + 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) + python_encoder = PolarEncoder( + block_size, + num_info_bits, + frozen_bit_positions, + frozen_bit_values) data = np.array([], dtype=int) ref = np.array([], dtype=int) @@ -85,7 +95,8 @@ class test_polar_encoder_systematic(gr_unittest.TestCase): d = np.random.randint(2, size=num_info_bits) data = np.append(data, d) ref = np.append(ref, python_encoder.encode_systematic(d)) - polar_encoder = fec.polar_encoder_systematic.make(block_size, num_info_bits, frozen_bit_positions) + polar_encoder = fec.polar_encoder_systematic.make( + block_size, num_info_bits, frozen_bit_positions) return data, ref, polar_encoder |