From d60005063ce734cc963e4a93e1406d22ccb1383b Mon Sep 17 00:00:00 2001 From: "Sybren A. St?vel" Date: Sun, 31 Jul 2011 19:54:53 +0200 Subject: Added yield_fixedblock --- tests/test_blocks.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'tests') diff --git a/tests/test_blocks.py b/tests/test_blocks.py index 8481c3c..ce5f03a 100644 --- a/tests/test_blocks.py +++ b/tests/test_blocks.py @@ -65,4 +65,11 @@ class VarblockTest(unittest.TestCase): varblocks = list(blocks.yield_varblocks(infile)) self.assertEqual(['12345', 'Sybren'], varblocks) +class FixedblockTest(unittest.TestCase): + def test_yield_fixedblock(self): + + infile = StringIO('123456Sybren') + + fixedblocks = list(blocks.yield_fixedblocks(infile, 6)) + self.assertEqual(['123456', 'Sybren'], fixedblocks) -- cgit v1.2.1