diff options
| author | Felix Wunsch <felix.wunsch@kit.edu> | 2015-08-13 13:21:52 +0200 |
|---|---|---|
| committer | Felix Wunsch <felix.wunsch@kit.edu> | 2015-08-13 13:39:38 +0200 |
| commit | 508c124479f05b76472f29330919abbd1b813c90 (patch) | |
| tree | 8404de63f111b9c9d7323750b4b48294f2e9282f /gr-blocks/python/blocks/qa_vector_insert.py | |
| parent | 22e2f1aed8afdfccce3884cf6bf3140c2b8e3f53 (diff) | |
blocks: make vector_insert propagate tags correctly
Diffstat (limited to 'gr-blocks/python/blocks/qa_vector_insert.py')
| -rwxr-xr-x | gr-blocks/python/blocks/qa_vector_insert.py | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/gr-blocks/python/blocks/qa_vector_insert.py b/gr-blocks/python/blocks/qa_vector_insert.py index e4c4055125..b916e3d528 100755 --- a/gr-blocks/python/blocks/qa_vector_insert.py +++ b/gr-blocks/python/blocks/qa_vector_insert.py @@ -53,6 +53,67 @@ class test_vector_insert(gr_unittest.TestCase): else: self.assertEqual(0, result_data[i]) + def test_002(self): # insert tags and check their propagation, zero offset + period = 11000 + offset = 0 + insert = [1.0,] * 1000 + + src = blocks.null_source(gr.sizeof_float) + s2ts = blocks.stream_to_tagged_stream(gr.sizeof_float, 1, period-len(insert), "packet") + head = blocks.head(gr.sizeof_float, 1000000) + ins = blocks.vector_insert_f(insert, period, offset) + dst = blocks.vector_sink_f() + + self.tb.connect(src, s2ts, head, ins, dst) + self.tb.run() + + expected_result = (1000, 12000, 23000, 34000, 45000, 56000, 67000) + tags = dst.tags() + offsets = [tag.offset for tag in tags] + for i in range(len(expected_result)): + self.assertTrue(expected_result[i] == offsets[i]) + + def test_003(self): # insert tags and check their propagation, non-zero offset + period = 11000 + offset = 1000 + insert = [1.0,] * 1000 + + src = blocks.null_source(gr.sizeof_float) + s2ts = blocks.stream_to_tagged_stream(gr.sizeof_float, 1, period-len(insert), "packet") + head = blocks.head(gr.sizeof_float, 1000000) + ins = blocks.vector_insert_f(insert, period, offset) + dst = blocks.vector_sink_f() + + self.tb.connect(src, s2ts, head, ins, dst) + self.tb.run() + + expected_result = (0, 11000, 22000, 33000, 44000, 55000, 66000) + tags = dst.tags() + offsets = [tag.offset for tag in tags] + for i in range(len(expected_result)): + self.assertTrue(expected_result[i] == offsets[i]) + + def test_004(self): # insert tags and check their propagation, non-zero offset, multiple tags per copy region + period = 11000 + offset = 1000 + packetlen = 2000 + insert = [1.0,] * 1000 + + src = blocks.null_source(gr.sizeof_float) + s2ts = blocks.stream_to_tagged_stream(gr.sizeof_float, 1, packetlen, "packet") + head = blocks.head(gr.sizeof_float, 1000000) + ins = blocks.vector_insert_f(insert, period, offset) + dst = blocks.vector_sink_f() + + self.tb.connect(src, s2ts, head, ins, dst) + self.tb.run() + + expected_result = (0, 2000, 4000, 6000, 8000, 11000, 13000, 15000, 17000, 19000, 22000, 24000, 26000) + tags = dst.tags() + offsets = [tag.offset for tag in tags] + for i in range(len(expected_result)): + self.assertTrue(expected_result[i] == offsets[i]) + if __name__ == '__main__': gr_unittest.run(test_vector_insert, "test_vector_insert.xml") |