diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/block_test.py | 44 |
1 files changed, 20 insertions, 24 deletions
diff --git a/tests/block_test.py b/tests/block_test.py index f7e18e4..ed6e97d 100644 --- a/tests/block_test.py +++ b/tests/block_test.py @@ -7,51 +7,49 @@ from demo_blocks import * class BlockTest(unittest.TestCase): + def setUp(self): + self.tb = gras.TopBlock() + + def tearDown(self): + self.tb = None + def test_vector_blocks(self): vec_source = VectorSource(numpy.uint32, [0, 9, 8, 7, 6]) vec_sink = VectorSink(numpy.uint32) - tb = gras.TopBlock() - tb.connect(vec_source, vec_sink) - tb.run() - tb = None + self.tb.connect(vec_source, vec_sink) + self.tb.run() self.assertEqual(vec_sink.get_vector(), (0, 9, 8, 7, 6)) def test_add_f32(self): - tb = gras.TopBlock() src0 = VectorSource(numpy.float32, [1, 3, 5, 7, 9]) src1 = VectorSource(numpy.float32, [0, 2, 4, 6, 8]) adder = Add2X(numpy.float32) sink = VectorSink(numpy.float32) - tb.connect((src0, 0), (adder, 0)) - tb.connect((src1, 0), (adder, 1)) - tb.connect(adder, sink) - tb.run() - tb = None + self.tb.connect((src0, 0), (adder, 0)) + self.tb.connect((src1, 0), (adder, 1)) + self.tb.connect(adder, sink) + self.tb.run() 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.tb.connect((src0, 0), (adder, 0)) + self.tb.connect((src1, 0), (adder, 1)) + self.tb.connect(adder, sink) + self.tb.run() self.assertEqual(sink.get_vector(), (1, 5j, 9, 13j, 17)) def test_tag_source_sink(self): values = (0, 'hello', 4.2, True, None, [2, 3, 4], (9, 8, 7), 1j, {2:'d'}) - tb = gras.TopBlock() src = TagSource(values) sink = TagSink() - tb.connect(src, sink) - tb.run() - tb = None + self.tb.connect(src, sink) + self.tb.run() self.assertEqual(sink.get_values(), values) def test_ro_buffers(self): @@ -69,10 +67,8 @@ class BlockTest(unittest.TestCase): source = VectorSource(numpy.uint32, [0, 9, 8, 7, 6]) sink = BadTouch(numpy.uint32) - tb = gras.TopBlock() - tb.connect(source, sink) - tb.run() - tb = None + self.tb.connect(source, sink) + self.tb.run() if __name__ == '__main__': unittest.main() |