More test cases

Author: cdce8p

mypyc/test-data/irbuild-str.test

@@ -145,6 +145,10 @@ def do_startswith(s1: str, s2: Tuple[str, ...]) -> bool:
 
 def do_endswith(s1: str, s2: Tuple[str, ...]) -> bool:
     return s1.endswith(s2)
+
+def do_tuple_literal_args(s1: str) -> None:
+    x = s1.startswith(("a", "b"))
+    y = s1.endswith(("a", "b"))
 [out]
 def do_startswith(s1, s2):
     s1 :: str
@@ -168,6 +172,37 @@ L0:
     r1 = r0 >= 0 :: signed
     r2 = truncate r0: i32 to builtins.bool
     return r2
+def do_tuple_literal_args(s1):
+    s1, r0, r1 :: str
+    r2 :: tuple[str, str]
+    r3 :: object
+    r4 :: i32
+    r5 :: bit
+    r6, x :: bool
+    r7, r8 :: str
+    r9 :: tuple[str, str]
+    r10 :: object
+    r11 :: i32
+    r12 :: bit
+    r13, y :: bool
+L0:
+    r0 = 'a'
+    r1 = 'b'
+    r2 = (r0, r1)
+    r3 = box(tuple[str, str], r2)
+    r4 = CPyStr_Startswith(s1, r3)
+    r5 = r4 >= 0 :: signed
+    r6 = truncate r4: i32 to builtins.bool
+    x = r6
+    r7 = 'a'
+    r8 = 'b'
+    r9 = (r7, r8)
+    r10 = box(tuple[str, str], r9)
+    r11 = CPyStr_Endswith(s1, r10)
+    r12 = r11 >= 0 :: signed
+    r13 = truncate r11: i32 to builtins.bool
+    y = r13
+    return 1
 
 [case testStrToBool]
 def is_true(x: str) -> bool:

mypyc/test-data/run-strings.test

@@ -22,11 +22,16 @@ def match(x: str, y: str) -> Tuple[bool, bool]:
     return (x.startswith(y), x.endswith(y))
 def match_tuple(x: str, y: Tuple[str, ...]) -> Tuple[bool, bool]:
     return (x.startswith(y), x.endswith(y))
+def match_tuple_literal_args(x: str, y: str, z: str) -> Tuple[bool, bool]:
+    return (x.startswith((y, z)), x.endswith((y, z)))
 def remove_prefix_suffix(x: str, y: str) -> Tuple[str, str]:
     return (x.removeprefix(y), x.removesuffix(y))
 
 [file driver.py]
-from native import f, g, tostr, booltostr, concat, eq, match, match_tuple, remove_prefix_suffix
+from native import (
+    f, g, tostr, booltostr, concat, eq, match, match_tuple,
+    match_tuple_literal_args, remove_prefix_suffix
+)
 import sys
 from testutil import assertRaises
 
@@ -58,6 +63,8 @@ with assertRaises(TypeError, "tuple for startswith must only contain str"):
     assert match_tuple('abc', (None,))
 with assertRaises(TypeError, "tuple for endswith must only contain str"):
     assert match_tuple('abc', ('a', None))
+assert match_tuple_literal_args('abc', 'z', 'a') == (True, False)
+assert match_tuple_literal_args('abc', 'z', 'c') == (False, True)
 
 assert remove_prefix_suffix('', '') == ('', '')
 assert remove_prefix_suffix('abc', 'a') == ('bc', 'abc')