[mypyc] Optimize str.__contains__

mypyc/doc/str_operations.rst

@@ -21,6 +21,7 @@ Operators
 * Slicing (``s[n:m]``, ``s[n:]``, ``s[:m]``)
 * Comparisons (``==``, ``!=``)
 * Augmented assignment (``s1 += s2``)
+* Containment (``s1 in s2``)
 
 .. _str-methods:
 

mypyc/primitives/str_ops.py

@@ -84,6 +84,17 @@
     error_kind=ERR_MAGIC,
 )
 
+# item in str
+binary_op(
+    name="in",
+    arg_types=[str_rprimitive, str_rprimitive],
+    return_type=c_int_rprimitive,
+    c_function_name="PyUnicode_Contains",
+    error_kind=ERR_NEG_INT,
+    truncated_type=bool_rprimitive,
+    ordering=[1, 0],
+)
+
 # str.join(obj)
 method_op(
     name="join",

mypyc/test-data/run-strings.test

@@ -140,11 +140,24 @@ def test_partition() -> None:
     with assertRaises(ValueError, "empty separator"):
         rpartition(s_partition, "")
 
+def contains(s: str, o: str) -> bool:
+    return o in s
+
 def getitem(s: str, index: int) -> str:
     return s[index]
 
 s = "abc"
 
+def test_contains() -> None:
+    assert contains(s, "a") is True
+    assert contains(s, "abc") is True
+    assert contains(s, "Hello") is False
+    assert contains(s, "bc") is True
+    assert contains(s, "abcd") is False
+    assert contains(s, "bb") is False
+    assert contains(s, "") is True
+    assert contains(s, " ") is False
+
 def test_getitem() -> None:
     assert getitem(s, 0) == "a"
     assert getitem(s, 1) == "b"