add split and rsplit

Author: jgirardet

tests/snippets/bytes.py

@@ -345,3 +345,181 @@
 assert b"www.example.com".lstrip(b"cmowz.") == b"example.com"
 assert b"   spacious   ".rstrip() == b"   spacious"
 assert b"mississippi".rstrip(b"ipz") == b"mississ"
+
+
+# split
+assert b"1,2,3".split(b",") == [b"1", b"2", b"3"]
+assert b"1,2,3".split(b",", maxsplit=1) == [b"1", b"2,3"]
+assert b"1,2,,3,".split(b",") == [b"1", b"2", b"", b"3", b""]
+assert b"1 2 3".split() == [b"1", b"2", b"3"]
+assert b"1 2 3".split(maxsplit=1) == [b"1", b"2 3"]
+assert b"   1   2   3   ".split() == [b"1", b"2", b"3"]
+assert b"k\ruh\nfz e f".split() == [b"k", b"uh", b"fz", b"e", b"f"]
+
+SPLIT_FIXTURES = [
+    [
+        [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3],
+        [4, 5],
+        [[1, 2, 3], [1, 2, 3], [1, 2, 3]],
+        [[1, 2, 3], [1, 2, 3], [1, 2, 3]],
+        -1,
+    ],
+    [
+        [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5],
+        [4, 5],
+        [[1, 2, 3], [1, 2, 3], [1, 2, 3], []],
+        [[1, 2, 3], [1, 2, 3], [1, 2, 3], []],
+        -1,
+    ],
+    [
+        [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 3],
+        [4, 5],
+        [[1, 2, 3], [1, 2, 3], [1, 2, 3], [3]],
+        [[1, 2, 3], [1, 2, 3], [1, 2, 3], [3]],
+        -1,
+    ],
+    [
+        [4, 5, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3],
+        [4, 5],
+        [[], [2, 3], [1, 2, 3], [1, 2, 3]],
+        [[], [2, 3], [1, 2, 3], [1, 2, 3]],
+        -1,
+    ],
+    [
+        [1, 4, 5, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3],
+        [4, 5],
+        [[1], [2, 3], [1, 2, 3], [1, 2, 3]],
+        [[1], [2, 3], [1, 2, 3], [1, 2, 3]],
+        -1,
+    ],
+    [
+        [1, 2, 3, 4, 5, 4, 5, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3],
+        [4, 5],
+        [[1, 2, 3], [], [], [1, 2, 3], [1, 2, 3]],
+        [[1, 2, 3], [], [], [1, 2, 3], [1, 2, 3]],
+        -1,
+    ],
+    # maxsplit
+    [
+        [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3],
+        [4, 5],
+        [[1, 2, 3], [1, 2, 3, 4, 5, 1, 2, 3]],
+        [[1, 2, 3, 4, 5, 1, 2, 3], [1, 2, 3]],
+        1,
+    ],
+    [
+        [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5],
+        [4, 5],
+        [[1, 2, 3], [1, 2, 3, 4, 5, 1, 2, 3, 4, 5]],
+        [[1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3], []],
+        1,
+    ],
+    [
+        [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 3],
+        [4, 5],
+        [[1, 2, 3], [1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 3]],
+        [[1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3], [3]],
+        1,
+    ],
+    [
+        [4, 5, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3],
+        [4, 5],
+        [[], [2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3]],
+        [[4, 5, 2, 3, 4, 5, 1, 2, 3], [1, 2, 3]],
+        1,
+    ],
+    [
+        [1, 4, 5, 2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3],
+        [4, 5],
+        [[1], [2, 3, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3]],
+        [[1, 4, 5, 2, 3, 4, 5, 1, 2, 3], [1, 2, 3]],
+        1,
+    ],
+    [
+        [1, 2, 3, 4, 5, 4, 5, 4, 5, 1, 2, 3, 4, 5, 1, 2, 3],
+        [4, 5],
+        [[1, 2, 3], [], [4, 5, 1, 2, 3, 4, 5, 1, 2, 3]],
+        [[1, 2, 3, 4, 5, 4, 5], [1, 2, 3], [1, 2, 3]],
+        2,
+    ],
+    [
+        [13, 13, 13, 117, 104, 10, 102, 122, 32, 101, 102, 9, 9],
+        None,
+        [[117, 104], [102, 122], [101, 102]],
+        [[117, 104], [102, 122], [101, 102]],
+        -1,
+    ],
+    [
+        [13, 13, 13, 117, 104, 10, 102, 122, 32, 101, 102, 9, 9],
+        None,
+        [[117, 104, 10, 102, 122, 32, 101, 102, 9, 9]],
+        [[13, 13, 13, 117, 104, 10, 102, 122, 32, 101, 102]],
+        0,
+    ],
+    [
+        [13, 13, 13, 117, 104, 10, 102, 122, 32, 101, 102, 9, 9],
+        None,
+        [[117, 104], [102, 122, 32, 101, 102, 9, 9]],
+        [[13, 13, 13, 117, 104, 10, 102, 122], [101, 102]],
+        1,
+    ],
+    [
+        [13, 13, 13, 117, 104, 10, 102, 122, 32, 101, 102, 9, 9],
+        None,
+        [[117, 104], [102, 122], [101, 102, 9, 9]],
+        [[13, 13, 13, 117, 104], [102, 122], [101, 102]],
+        2,
+    ],
+    [
+        [13, 13, 13, 117, 104, 10, 10, 10, 102, 122, 32, 32, 101, 102, 9, 9],
+        None,
+        [[117, 104], [102, 122], [101, 102]],
+        [[117, 104], [102, 122], [101, 102]],
+        -1,
+    ],
+    [[49, 44, 50, 44, 51], [44], [[49], [50], [51]], [[49], [50], [51]], -1],
+    [[49, 44, 50, 44, 51], [44], [[49], [50, 44, 51]], [[49, 44, 50], [51]], 1],
+    [
+        [49, 44, 50, 44, 44, 51, 44],
+        [44],
+        [[49], [50], [], [51], []],
+        [[49], [50], [], [51], []],
+        -1,
+    ],
+    [[49, 32, 50, 32, 51], None, [[49], [50], [51]], [[49], [50], [51]], -1],
+    [[49, 32, 50, 32, 51], None, [[49], [50, 32, 51]], [[49, 32, 50], [51]], 1],
+    [
+        [32, 32, 32, 49, 32, 32, 32, 50, 32, 32, 32, 51, 32, 32, 32],
+        None,
+        [[49], [50], [51]],
+        [[49], [50], [51]],
+        -1,
+    ],
+]
+
+
+# for i in SPLIT_FIXTURES:  # for not yet implemented : TypeError: Unsupported method: __next__
+n_sp = 0
+while n_sp < len(SPLIT_FIXTURES):
+    i = SPLIT_FIXTURES[n_sp]
+    sep = None if i[1] == None else bytes(i[1])
+    try:
+        assert bytes(i[0]).split(sep=sep, maxsplit=i[4]) == [bytes(j) for j in i[2]]
+    except AssertionError:
+        print(i[0], i[1], i[2])
+        print(
+            "Expected : ", [list(x) for x in bytes(i[0]).split(sep=sep, maxsplit=i[4])]
+        )
+        break
+
+    try:
+        assert bytes(i[0]).rsplit(sep=sep, maxsplit=i[4]) == [bytes(j) for j in i[3]]
+    except AssertionError:
+        print(i[0], i[1], i[2])
+        print(
+            "Expected Rev : ",
+            [list(x) for x in bytes(i[0]).rsplit(sep=sep, maxsplit=i[4])],
+        )
+        break
+
+    n_sp += 1