py: In string.count, handle case of zero-length needle.

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/basics/string_count.py b/tests/basics/string_count.py
index bac99e78d8..0da1b1fcae 100644
--- a/tests/basics/string_count.py
+++ b/tests/basics/string_count.py
@@ -1,3 +1,29 @@
+print("".count(""))
+print("".count("a"))
+print("a".count(""))
+print("a".count("a"))
+print("a".count("b"))
+print("b".count("a"))
+
+print("aaa".count(""))
+print("aaa".count("a"))
+print("aaa".count("aa"))
+print("aaa".count("aaa"))
+print("aaa".count("aaaa"))
+
+print("aaaa".count(""))
+print("aaaa".count("a"))
+print("aaaa".count("aa"))
+print("aaaa".count("aaa"))
+print("aaaa".count("aaaa"))
+print("aaaa".count("aaaaa"))
+
+print("aaa".count("", 1))
+print("aaa".count("", 2))
+print("aaa".count("", 3))
+
+print("aaa".count("", 1, 2))
+
 print("asdfasdfaaa".count("asdf", -100))
 print("asdfasdfaaa".count("asdf", -8))
 print("asdf".count('s', True))