summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorSybren A. St?vel <sybren@stuvel.eu>2011-07-31 19:54:53 +0200
committerSybren A. St?vel <sybren@stuvel.eu>2011-07-31 19:54:53 +0200
commitd60005063ce734cc963e4a93e1406d22ccb1383b (patch)
tree53093b5e0367185d93f367542abfcfc2c6f49f78 /tests
parent782f831e8d4f0a02250a5457e96040f1fcf56b36 (diff)
downloadrsa-d60005063ce734cc963e4a93e1406d22ccb1383b.tar.gz
Added yield_fixedblock
Diffstat (limited to 'tests')
-rw-r--r--tests/test_blocks.py7
1 files changed, 7 insertions, 0 deletions
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)