py: Make bytes objs work with more str methods; add tests.

1 files changed, 28 insertions, 0 deletions

diff --git a/tests/basics/bytes_split.py b/tests/basics/bytes_split.py
new file mode 100644
index 0000000000..a9dda1ee8b
--- /dev/null
+++ b/tests/basics/bytes_split.py
@@ -0,0 +1,28 @@
+# default separator (whitespace)
+print(b"a b".split())
+print(b"   a   b    ".split(None))
+print(b"   a   b    ".split(None, 1))
+print(b"   a   b    ".split(None, 2))
+print(b"   a   b  c  ".split(None, 1))
+print(b"   a   b  c  ".split(None, 0))
+print(b"   a   b  c  ".split(None, -1))
+
+# empty separator should fail
+try:
+    b"abc".split(b'')
+except ValueError:
+    print("ValueError")
+
+# non-empty separator
+print(b"abc".split(b"a"))
+print(b"abc".split(b"b"))
+print(b"abc".split(b"c"))
+print(b"abc".split(b"z"))
+print(b"abc".split(b"ab"))
+print(b"abc".split(b"bc"))
+print(b"abc".split(b"abc"))
+print(b"abc".split(b"abcd"))
+print(b"abcabc".split(b"bc"))
+print(b"abcabc".split(b"bc", 0))
+print(b"abcabc".split(b"bc", 1))
+print(b"abcabc".split(b"bc", 2))