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-blocks/python/blocks/qa_python_message_passing.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-blocks/python/blocks/qa_python_message_passing.py')
| -rw-r--r-- | gr-blocks/python/blocks/qa_python_message_passing.py | 30 |
1 files changed, 20 insertions, 10 deletions
diff --git a/gr-blocks/python/blocks/qa_python_message_passing.py b/gr-blocks/python/blocks/qa_python_message_passing.py index e5b0bc6d9f..d36145793a 100644 --- a/gr-blocks/python/blocks/qa_python_message_passing.py +++ b/gr-blocks/python/blocks/qa_python_message_passing.py @@ -15,20 +15,21 @@ import numpy import time # Simple block to generate messages + + class message_generator(gr.sync_block): def __init__(self, msg_list, msg_interval): gr.sync_block.__init__( self, - name = "message generator", - in_sig = [numpy.float32], - out_sig = None + name="message generator", + in_sig=[numpy.float32], + out_sig=None ) self.msg_list = msg_list self.msg_interval = msg_interval self.msg_ctr = 0 self.message_port_register_out(pmt.intern('out_port')) - def work(self, input_items, output_items): inLen = len(input_items[0]) while self.msg_ctr < len(self.msg_list) and \ @@ -40,13 +41,15 @@ class message_generator(gr.sync_block): return inLen # Simple block to consume messages + + class message_consumer(gr.sync_block): def __init__(self): gr.sync_block.__init__( self, - name = "message consumer", - in_sig = None, - out_sig = None + name="message consumer", + in_sig=None, + out_sig=None ) self.msg_list = [] self.message_port_register_in(pmt.intern('in_port')) @@ -57,6 +60,7 @@ class message_consumer(gr.sync_block): # Create a new PMT from long value and put in list self.msg_list.append(pmt.from_long(pmt.to_long(msg))) + class test_python_message_passing(gr_unittest.TestCase): def setUp(self): @@ -74,7 +78,7 @@ class test_python_message_passing(gr_unittest.TestCase): # Create vector source with dummy data to trigger messages src_data = [] - for i in range(num_msgs*msg_interval): + for i in range(num_msgs * msg_interval): src_data.append(float(i)) src = blocks.vector_source_f(src_data, False) msg_gen = message_generator(msg_list, msg_interval) @@ -87,8 +91,13 @@ class test_python_message_passing(gr_unittest.TestCase): self.tb.msg_connect(msg_gen, 'out_port', msg_cons, 'in_port') # Verify that the messgae port query functions work - self.assertEqual(pmt.to_python(msg_gen.message_ports_out())[0], 'out_port') - self.assertEqual('in_port' in pmt.to_python(msg_cons.message_ports_in()), True) + self.assertEqual( + pmt.to_python( + msg_gen.message_ports_out())[0], + 'out_port') + self.assertEqual( + 'in_port' in pmt.to_python( + msg_cons.message_ports_in()), True) # Run to verify message passing self.tb.run() @@ -98,5 +107,6 @@ class test_python_message_passing(gr_unittest.TestCase): for i in range(num_msgs): self.assertTrue(pmt.equal(msg_list[i], msg_cons.msg_list[i])) + if __name__ == '__main__': gr_unittest.run(test_python_message_passing) |