diff options
Diffstat (limited to 'tests/block_test.py')
| -rw-r--r-- | tests/block_test.py | 23 |
1 files changed, 22 insertions, 1 deletions
diff --git a/tests/block_test.py b/tests/block_test.py index 23322a4..6d922a9 100644 --- a/tests/block_test.py +++ b/tests/block_test.py @@ -21,7 +21,7 @@ class BlockTest(unittest.TestCase): tb = gras.TopBlock() src0 = VectorSource(numpy.float32, [1, 3, 5, 7, 9]) src1 = VectorSource(numpy.float32, [0, 2, 4, 6, 8]) - adder = add_2x(numpy.float32) + adder = Add2X(numpy.float32) sink = VectorSink(numpy.float32) tb.connect((src0, 0), (adder, 0)) tb.connect((src1, 0), (adder, 1)) @@ -30,5 +30,26 @@ class BlockTest(unittest.TestCase): tb = None self.assertEqual(sink.get_vector(), (1, 5, 9, 13, 17)) + def test_add_fc32(self): + tb = gras.TopBlock() + src0 = VectorSource(numpy.complex64, [1, 3j, 5, 7j, 9]) + src1 = VectorSource(numpy.complex64, [0, 2j, 4, 6j, 8]) + adder = Add2X(numpy.complex64) + sink = VectorSink(numpy.complex64) + tb.connect((src0, 0), (adder, 0)) + tb.connect((src1, 0), (adder, 1)) + tb.connect(adder, sink) + tb.run() + tb = None + self.assertEqual(sink.get_vector(), (1, 5j, 9, 13j, 17)) + + def test_tag_source(self): + tb = gras.TopBlock() + src = TagSource([1, 2, 3]) + sink = NullSink(numpy.uint8) + tb.connect(src, sink) + tb.run() + tb = None + if __name__ == '__main__': unittest.main() |