diff options
Diffstat (limited to 'gr-digital/python/digital/qa_diff_encoder.py')
| -rw-r--r-- | gr-digital/python/digital/qa_diff_encoder.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/gr-digital/python/digital/qa_diff_encoder.py b/gr-digital/python/digital/qa_diff_encoder.py index 15d80e5281..7ffc331b47 100644 --- a/gr-digital/python/digital/qa_diff_encoder.py +++ b/gr-digital/python/digital/qa_diff_encoder.py @@ -13,6 +13,7 @@ import random from gnuradio import gr, gr_unittest, digital, blocks + def make_random_int_list(L, min, max): result = [] for x in range(L): @@ -32,7 +33,7 @@ class test_diff_encoder(gr_unittest.TestCase): def test_diff_encdec_000(self): random.seed(0) modulus = 2 - src_data = make_random_int_list(1000, 0, modulus-1) + src_data = make_random_int_list(1000, 0, modulus - 1) expected_result = src_data src = blocks.vector_source_b(src_data) enc = digital.diff_encoder_bb(modulus) @@ -46,7 +47,7 @@ class test_diff_encoder(gr_unittest.TestCase): def test_diff_encdec_001(self): random.seed(0) modulus = 4 - src_data = make_random_int_list(1000, 0, modulus-1) + src_data = make_random_int_list(1000, 0, modulus - 1) expected_result = src_data src = blocks.vector_source_b(src_data) enc = digital.diff_encoder_bb(modulus) @@ -60,7 +61,7 @@ class test_diff_encoder(gr_unittest.TestCase): def test_diff_encdec_002(self): random.seed(0) modulus = 8 - src_data = make_random_int_list(40000, 0, modulus-1) + src_data = make_random_int_list(40000, 0, modulus - 1) expected_result = src_data src = blocks.vector_source_b(src_data) enc = digital.diff_encoder_bb(modulus) @@ -71,6 +72,6 @@ class test_diff_encoder(gr_unittest.TestCase): actual_result = dst.data() # fetch the contents of the sink self.assertEqual(expected_result, actual_result) + if __name__ == '__main__': gr_unittest.run(test_diff_encoder) - |