implement 'util.advance()'

2017-12-03 01:38:24 +01:00
parent 0e5057b15d
commit 93482a1f88
3 changed files with 28 additions and 13 deletions
--- a/test/test_util.py
+++ b/test/test_util.py
@@ -205,6 +205,18 @@ class TestOther(unittest.TestCase):
        self.assertEqual(util.parse_bytes("invalid"), 0)
        self.assertEqual(util.parse_bytes(" 123 kb "), 0)

+    def test_advance(self):
+        items = range(5)
+
+        self.assertCountEqual(
+            util.advance(items, 0), items)
+        self.assertCountEqual(
+            util.advance(items, 3), range(3, 5))
+        self.assertCountEqual(
+            util.advance(items, 9), [])
+        self.assertCountEqual(
+            util.advance(util.advance(items, 1), 2), range(3, 5))
+
    def test_combine_dict(self):
        self.assertEqual(
            util.combine_dict({}, {}),