diff options
Diffstat (limited to 'gr-pdu/python/pdu/qa_take_skip_to_pdu.py')
-rwxr-xr-x | gr-pdu/python/pdu/qa_take_skip_to_pdu.py | 64 |
1 files changed, 34 insertions, 30 deletions
diff --git a/gr-pdu/python/pdu/qa_take_skip_to_pdu.py b/gr-pdu/python/pdu/qa_take_skip_to_pdu.py index 76414c857a..bcc33e6f4f 100755 --- a/gr-pdu/python/pdu/qa_take_skip_to_pdu.py +++ b/gr-pdu/python/pdu/qa_take_skip_to_pdu.py @@ -14,6 +14,7 @@ import pmt import time import numpy + class qa_take_skip_to_pdu_X (gr_unittest.TestCase): # this method is necessary because by default pmt.equal does not evaluate @@ -34,67 +35,70 @@ class qa_take_skip_to_pdu_X (gr_unittest.TestCase): print("vectors not equal? " + repr(vec1) + repr(vec2)) self.assertTrue(False) - def setUp (self): - self.tb = gr.top_block () + def setUp(self): + self.tb = gr.top_block() - def tearDown (self): + def tearDown(self): self.tb = None - def test_001_f_32 (self): - self.source = blocks.vector_source_f(range(0,32*3), False, 1, []) + def test_001_f_32(self): + self.source = blocks.vector_source_f(range(0, 32 * 3), False, 1, []) self.ts_pdu = pdu.take_skip_to_pdu_f(32, 32) self.debug = blocks.message_debug() self.tb.connect((self.source, 0), (self.ts_pdu, 0)) self.tb.msg_connect((self.ts_pdu, 'pdus'), (self.debug, 'store')) - dic = pmt.dict_add(pmt.make_dict(), pmt.intern("pdu_num"), pmt.from_uint64(0)) - vec = pmt.init_f32vector(32, range(0,32)) - expected = pmt.cons(dic,vec) - self.tb.run () + dic = pmt.dict_add(pmt.make_dict(), pmt.intern( + "pdu_num"), pmt.from_uint64(0)) + vec = pmt.init_f32vector(32, range(0, 32)) + expected = pmt.cons(dic, vec) + self.tb.run() actual = self.debug.get_message(0) self.assertEqualPDU(actual, expected) - def test_002_c_80 (self): - self.source = blocks.vector_source_c(range(0,32*3), False, 1, []) + def test_002_c_80(self): + self.source = blocks.vector_source_c(range(0, 32 * 3), False, 1, []) self.ts_pdu = pdu.take_skip_to_pdu_c(80, 32) self.debug = blocks.message_debug() self.tb.connect((self.source, 0), (self.ts_pdu, 0)) self.tb.msg_connect((self.ts_pdu, 'pdus'), (self.debug, 'store')) - dic = pmt.dict_add(pmt.make_dict(), pmt.intern("pdu_num"), pmt.from_uint64(0)) - vec = pmt.init_c32vector(80, range(0,80)) - expected = pmt.cons(dic,vec) - self.tb.run () + dic = pmt.dict_add(pmt.make_dict(), pmt.intern( + "pdu_num"), pmt.from_uint64(0)) + vec = pmt.init_c32vector(80, range(0, 80)) + expected = pmt.cons(dic, vec) + self.tb.run() actual = self.debug.get_message(0) self.assertEqualPDU(actual, expected) - - def test_003_s_2_11_7 (self): - self.source = blocks.vector_source_s(range(0,32*3), False, 1, []) + def test_003_s_2_11_7(self): + self.source = blocks.vector_source_s(range(0, 32 * 3), False, 1, []) self.ts_pdu = pdu.take_skip_to_pdu_s(2, 11) self.debug = blocks.message_debug() self.tb.connect((self.source, 0), (self.ts_pdu, 0)) self.tb.msg_connect((self.ts_pdu, 'pdus'), (self.debug, 'store')) - dic = pmt.dict_add(pmt.make_dict(), pmt.intern("pdu_num"), pmt.from_uint64(7)) - vec = pmt.init_s16vector(2, list(range(91,93))) - expected = pmt.cons(dic,vec) - self.tb.run () + dic = pmt.dict_add(pmt.make_dict(), pmt.intern( + "pdu_num"), pmt.from_uint64(7)) + vec = pmt.init_s16vector(2, list(range(91, 93))) + expected = pmt.cons(dic, vec) + self.tb.run() actual = self.debug.get_message(7) self.assertEqualPDU(actual, expected) - - def test_004_b_512 (self): - self.source = blocks.vector_source_b(list(range(0,256))*4, False, 1, []) - self.ts_pdu = pdu.take_skip_to_pdu_b(512,1) + def test_004_b_512(self): + self.source = blocks.vector_source_b( + list(range(0, 256)) * 4, False, 1, []) + self.ts_pdu = pdu.take_skip_to_pdu_b(512, 1) self.debug = blocks.message_debug() self.tb.connect((self.source, 0), (self.ts_pdu, 0)) self.tb.msg_connect((self.ts_pdu, 'pdus'), (self.debug, 'store')) - dic = pmt.dict_add(pmt.make_dict(), pmt.intern("pdu_num"), pmt.from_uint64(0)) - vec = pmt.init_u8vector(512, list(range(0,256))*2) - expected = pmt.cons(dic,vec) - self.tb.run () + dic = pmt.dict_add(pmt.make_dict(), pmt.intern( + "pdu_num"), pmt.from_uint64(0)) + vec = pmt.init_u8vector(512, list(range(0, 256)) * 2) + expected = pmt.cons(dic, vec) + self.tb.run() actual = self.debug.get_message(0) self.assertEqualPDU(actual, expected) |