limiting the number of excursion in an iterator

mila-iqia · Jun 24, 2015 · 88ccff7 · 88ccff7
1 parent eb1e7bd
commit 88ccff7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_schemes.py b/tests/test_schemes.py
@@ -102,6 +102,7 @@ def test_concatenated_scheme():
     sch = ConcatenatedScheme(schemes=[ConstantScheme(batch_size=10, times=5),
                                       ConstantScheme(batch_size=20, times=3),
                                       ConstantScheme(batch_size=30, times=1)])
+
     assert (list(sch.get_request_iterator()) ==
             ([10] * 5) + ([20] * 3) + [30])