[Commit-gnuradio] [gnuradio] 26/46: zeromq: Add qa tests

[git]

This is an automated email from the git hooks/post-receive script.

jcorgan pushed a commit to branch master
in repository gnuradio.

commit a036dd2f3e74387316946fbac47282a9e7c1e87d
Author: Johannes Schmitz <address@hidden>
Date:   Wed May 7 18:00:52 2014 +0200

    zeromq: Add qa tests
---
 gr-zeromq/python/zeromq/qa_zeromq_pushpull.py | 50 +++++++++++++++++++++++++++
 gr-zeromq/python/zeromq/qa_zeromq_reqrep.py   | 50 +++++++++++++++++++++++++++
 2 files changed, 100 insertions(+)

diff --git a/gr-zeromq/python/zeromq/qa_zeromq_pushpull.py 
b/gr-zeromq/python/zeromq/qa_zeromq_pushpull.py
new file mode 100755
index 0000000..72c024e
--- /dev/null
+++ b/gr-zeromq/python/zeromq/qa_zeromq_pushpull.py
@@ -0,0 +1,50 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# Copyright 202013 Free Software Foundation, Inc.
+#
+# This file is part of GNU Radio
+#
+# GNU Radio is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3, or (at your option)
+# any later version.
+#
+# GNU Radio is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with GNU Radio; see the file COPYING.  If not, write to
+# the Free Software Foundation, Inc., 51 Franklin Street,
+# Boston, MA 02110-1301, USA.
+#
+
+from gnuradio import gr, gr_unittest
+from gnuradio import blocks, zeromq
+from gnuradio import eng_notation
+
+class qa_zeromq_pushpull (gr_unittest.TestCase):
+
+    def setUp (self):
+        self.tb = gr.top_block ()
+
+    def tearDown (self):
+        self.tb = None
+
+    def test_001_t (self):
+        src_data = [1,2,3,4,5,6,7,8,9]*100
+        print src_data
+        src = blocks.vector_source_c(src_data, False, 1)
+        zeromq_push_sink = zeromq.push_sink(gr.sizeof_gr_complex, 1, 
"tcp://*:5555", True)
+        zeromq_pull_source = zeromq.pull_source(gr.sizeof_gr_complex, 1, 
"tcp://localhost:5555", 0.1)
+        sink = blocks.vector_sink_c()
+#        print sink.data
+        self.tb.connect(src, zeromq_push_sink)
+        self.tb.connect(zeromq_pull_source, sink)
+        self.tb.run ()
+        self.assertEqual(sink.data(), src_data)
+
+if __name__ == '__main__':
+    gr_unittest.run(qa_zeromq_pushpull, "qa_zeromq_pushpull.xml")
diff --git a/gr-zeromq/python/zeromq/qa_zeromq_reqrep.py 
b/gr-zeromq/python/zeromq/qa_zeromq_reqrep.py
new file mode 100755
index 0000000..e520472
--- /dev/null
+++ b/gr-zeromq/python/zeromq/qa_zeromq_reqrep.py
@@ -0,0 +1,50 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# Copyright 202013 Free Software Foundation, Inc.
+#
+# This file is part of GNU Radio
+#
+# GNU Radio is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3, or (at your option)
+# any later version.
+#
+# GNU Radio is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with GNU Radio; see the file COPYING.  If not, write to
+# the Free Software Foundation, Inc., 51 Franklin Street,
+# Boston, MA 02110-1301, USA.
+#
+
+from gnuradio import gr, gr_unittest
+from gnuradio import blocks, zeromq
+from gnuradio import eng_notation
+
+class qa_zeromq_reqrep (gr_unittest.TestCase):
+
+    def setUp (self):
+        self.tb = gr.top_block ()
+
+    def tearDown (self):
+        self.tb = None
+
+    def test_001_t (self):
+        src_data = [1,2,3,4,5,6,7,8,9]*100
+        print src_data
+        src = blocks.vector_source_c(src_data, False, 1)
+        zeromq_rep_sink = zeromq.rep_sink(gr.sizeof_gr_complex, 1, 
"tcp://*:5555", 0.1, True)
+        zeromq_req_source = zeromq.req_source(gr.sizeof_gr_complex, 1, 
"tcp://localhost:5555")
+        sink = blocks.vector_sink_c()
+#        print sink.data
+        self.tb.connect(src, zeromq_rep_sink)
+        self.tb.connect(zeromq_req_source, sink)
+        self.tb.run ()
+        self.assertEqual(sink.data(), src_data)
+
+if __name__ == '__main__':
+    gr_unittest.run(qa_zeromq_reqrep, "qa_zeromq_reqrep.xml")