working on tags python support

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()