[Commit-gnuradio] [gnuradio] 04/09: blocks: added tag tests to vector source QA code

[git]

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

jcorgan pushed a commit to branch maint
in repository gnuradio.

commit 79b850208ddaa13b474a2d32e291bc67cb613123
Author: Sean Nowlan <address@hidden>
Date:   Sun Apr 26 17:05:16 2015 -0400

    blocks: added tag tests to vector source QA code
---
 gr-blocks/python/blocks/qa_vector_sink_source.py | 52 ++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/gr-blocks/python/blocks/qa_vector_sink_source.py 
b/gr-blocks/python/blocks/qa_vector_sink_source.py
index 6c9f933..5dab701 100755
--- a/gr-blocks/python/blocks/qa_vector_sink_source.py
+++ b/gr-blocks/python/blocks/qa_vector_sink_source.py
@@ -21,8 +21,22 @@
 #
 
 from gnuradio import gr, gr_unittest, blocks
+import pmt
 import math
 
+def make_tag(key, value, offset, srcid=None):
+    tag = gr.tag_t()
+    tag.key = pmt.string_to_symbol(key)
+    tag.value = pmt.to_pmt(value)
+    tag.offset = offset
+    if srcid is not None:
+        tag.srcid = pmt.to_pmt(srcid)
+    return tag
+
+def compare_tags(a, b):
+    return a.offset == b.offset and pmt.equal(a.key, b.key) and \
+           pmt.equal(a.value, b.value) and pmt.equal(a.srcid, b.srcid)
+
 class test_vector_sink_source(gr_unittest.TestCase):
 
     def setUp(self):
@@ -60,6 +74,44 @@ class test_vector_sink_source(gr_unittest.TestCase):
         expected_result = tuple(src_data)
         self.assertRaises(RuntimeError, lambda : 
blocks.vector_source_f(src_data, False, 3))
 
+    def test_004(self):
+        src_data = [float(x) for x in range(16)]
+        expected_result = tuple(src_data)
+        src_tags = tuple([make_tag('key', 'val', 0, 'src')])
+        expected_tags = src_tags[:]
+
+        src = blocks.vector_source_f(src_data, repeat=False, tags=src_tags)
+        dst = blocks.vector_sink_f()
+
+        self.tb.connect(src, dst)
+        self.tb.run()
+        result_data = dst.data()
+        result_tags = dst.tags()
+        self.assertEqual(expected_result, result_data)
+        self.assertEqual(len(result_tags), 1)
+        self.assertTrue(compare_tags(expected_tags[0], result_tags[0]))
+
+    def test_005(self):
+        length = 16
+        src_data = [float(x) for x in range(length)]
+        expected_result = tuple(src_data + src_data)
+        src_tags = tuple([make_tag('key', 'val', 0, 'src')])
+        expected_tags = tuple([make_tag('key', 'val', 0, 'src'),
+                               make_tag('key', 'val', length, 'src')])
+
+        src = blocks.vector_source_f(src_data, repeat=True, tags=src_tags)
+        head = blocks.head(gr.sizeof_float, 2*length)
+        dst = blocks.vector_sink_f()
+
+        self.tb.connect(src, head, dst)
+        self.tb.run()
+        result_data = dst.data()
+        result_tags = dst.tags()
+        self.assertEqual(expected_result, result_data)
+        self.assertEqual(len(result_tags), 2)
+        self.assertTrue(compare_tags(expected_tags[0], result_tags[0]))
+        self.assertTrue(compare_tags(expected_tags[1], result_tags[1]))
+
 if __name__ == '__main__':
     gr_unittest.run(test_vector_sink_source, "test_vector_sink_source.xml")