add test for desired feature

hugapi · Dec 1, 2015 · 0bd20dd · 0bd20dd
1 parent e7b71b8
commit 0bd20dd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_output_format.py b/tests/test_output_format.py
@@ -66,6 +66,9 @@ class NamedTupleObject(namedtuple('BaseTuple', ('name', 'value'))):
     data = set((1, 2, 3, 3))
     assert hug.input_format.json(BytesIO(hug.output_format.json(data))) == [1, 2, 3]
 
+    data = (number for number in range(1, 4))
+    assert hug.input_format.json(BytesIO(hug.output_format.json(data))) == [1, 2, 3]
+
     with open('README.md', 'rb') as json_file:
         assert hasattr(hug.output_format.json(json_file), 'read')