Merge pull request #351 from xbe/str-partition

Implement str.partition and add tests for it.

2 files changed, 58 insertions, 0 deletions

diff --git a/tests/basics/string_partition.py b/tests/basics/string_partition.py
new file mode 100644
index 0000000000..ad70d02509
--- /dev/null
+++ b/tests/basics/string_partition.py
@@ -0,0 +1,29 @@
+print("asdf".partition('g'))
+print("asdf".partition('a'))
+print("asdf".partition('s'))
+print("asdf".partition('f'))
+print("asdf".partition('d'))
+print("asdf".partition('asd'))
+print("asdf".partition('sdf'))
+print("asdf".partition('as'))
+print("asdf".partition('df'))
+print("asdf".partition('asdf'))
+print("asdf".partition('asdfa'))
+print("asdf".partition('fasdf'))
+print("asdf".partition('fasdfa'))
+print("abba".partition('a'))
+print("abba".partition('b'))
+
+try:
+    print("asdf".partition(1))
+except TypeError:
+    print("Raised TypeError")
+else:
+    print("Did not raise TypeError")
+
+try:
+    print("asdf".partition(''))
+except ValueError:
+    print("Raised ValueError")
+else:
+    print("Did not raise ValueError")
diff --git a/tests/basics/string_rpartition.py b/tests/basics/string_rpartition.py
new file mode 100644
index 0000000000..656121c94d
--- /dev/null
+++ b/tests/basics/string_rpartition.py
@@ -0,0 +1,29 @@
+print("asdf".rpartition('g'))
+print("asdf".rpartition('a'))
+print("asdf".rpartition('s'))
+print("asdf".rpartition('f'))
+print("asdf".rpartition('d'))
+print("asdf".rpartition('asd'))
+print("asdf".rpartition('sdf'))
+print("asdf".rpartition('as'))
+print("asdf".rpartition('df'))
+print("asdf".rpartition('asdf'))
+print("asdf".rpartition('asdfa'))
+print("asdf".rpartition('fasdf'))
+print("asdf".rpartition('fasdfa'))
+print("abba".rpartition('a'))
+print("abba".rpartition('b'))
+
+try:
+    print("asdf".rpartition(1))
+except TypeError:
+    print("Raised TypeError")
+else:
+    print("Did not raise TypeError")
+
+try:
+    print("asdf".rpartition(''))
+except ValueError:
+    print("Raised ValueError")
+else:
+    print("Did not raise ValueError")