diff options
| author | Johnathan Corgan <johnathan@corganlabs.com> | 2015-08-30 13:53:39 -0700 |
|---|---|---|
| committer | Johnathan Corgan <johnathan@corganlabs.com> | 2015-08-30 13:53:39 -0700 |
| commit | a06420691493534ca268ce52e1f16504c216828d (patch) | |
| tree | d5cd78e9612f1ab011d636761f5077982ee7b13b /gr-blocks/python | |
| parent | 4176b1be3ffef88924eaedfc1238aa653e8496a8 (diff) | |
| parent | 508c124479f05b76472f29330919abbd1b813c90 (diff) | |
Merge remote-tracking branch 'fewu/insert_vector_tag_fixes'
Diffstat (limited to 'gr-blocks/python')
| -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") |