diff options
Diffstat (limited to 'gr-zeromq/python/zeromq/qa_zeromq_pushpull.py')
| -rwxr-xr-x | gr-zeromq/python/zeromq/qa_zeromq_pushpull.py | 20 |
1 files changed, 13 insertions, 7 deletions
diff --git a/gr-zeromq/python/zeromq/qa_zeromq_pushpull.py b/gr-zeromq/python/zeromq/qa_zeromq_pushpull.py index d09b92c130..388f7ff9cb 100755 --- a/gr-zeromq/python/zeromq/qa_zeromq_pushpull.py +++ b/gr-zeromq/python/zeromq/qa_zeromq_pushpull.py @@ -26,10 +26,12 @@ import time class qa_zeromq_pushpull (gr_unittest.TestCase): def setUp (self): - self.tb = gr.top_block () + self.send_tb = gr.top_block() + self.recv_tb = gr.top_block() def tearDown (self): - self.tb = None + self.send_tb = None + self.recv_tb = None def test_001 (self): vlen = 10 @@ -38,12 +40,16 @@ class qa_zeromq_pushpull (gr_unittest.TestCase): zeromq_push_sink = zeromq.push_sink(gr.sizeof_float, vlen, "tcp://127.0.0.1:5557") zeromq_pull_source = zeromq.pull_source(gr.sizeof_float, vlen, "tcp://127.0.0.1:5557", 0) sink = blocks.vector_sink_f(vlen) - self.tb.connect(src, zeromq_push_sink) - self.tb.connect(zeromq_pull_source, sink) - self.tb.start() + self.send_tb.connect(src, zeromq_push_sink) + self.recv_tb.connect(zeromq_pull_source, sink) + self.recv_tb.start() time.sleep(0.25) - self.tb.stop() - self.tb.wait() + self.send_tb.start() + time.sleep(0.25) + self.recv_tb.stop() + self.send_tb.stop() + self.recv_tb.wait() + self.send_tb.wait() self.assertFloatTuplesAlmostEqual(sink.data(), src_data) if __name__ == '__main__': |