From 2091d1638734022a9ff879148ed432556c99db0f Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Sat, 2 May 2015 17:19:48 +1200 Subject: test coverage: generators and websockets to 100% --- test/test_language_generators.py | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'test/test_language_generators.py') diff --git a/test/test_language_generators.py b/test/test_language_generators.py index 3db1ac91..854b8485 100644 --- a/test/test_language_generators.py +++ b/test/test_language_generators.py @@ -36,3 +36,14 @@ def test_filegenerator(): # remove all references to FileGenerator instance to close the file # handle. del g + + +def test_transform_generator(): + def trans(offset, data): + return "a" * len(data) + g = generators.LiteralGenerator("one") + t = generators.TransformGenerator(g, trans) + assert len(t) == len(g) + assert t[0] == "a" + assert t[:] == "a" * len(g) + assert repr(t) -- cgit v1.2.3