feat(test): test containers from literal exprs

Author: BobTheBuidler

mypyc/test-data/irbuild-lists.test

@@ -620,6 +620,47 @@ L4:
     a = r3
     return 1
 
+[case testListBuiltFromStrExpr]
+def f2(val: str) -> str:
+    return val + "f2"
+
+def test() -> None:
+    a = [f2(x) for x in "abc"]
+[out]
+def f2(val):
+    val, r0, r1 :: str
+L0:
+    r0 = 'f2'
+    r1 = PyUnicode_Concat(val, r0)
+    return r1
+def test():
+    r0 :: str
+    r1 :: list
+    r2 :: native_int
+    r3 :: bit
+    r4, x, r5 :: str
+    r6 :: native_int
+    a :: list
+L0:
+    r0 = 'abc'
+    r1 = PyList_New(3)
+    r2 = 0
+L1:
+    r3 = r2 < 3 :: signed
+    if r3 goto L2 else goto L4 :: bool
+L2:
+    r4 = CPyStr_GetItemUnsafe(r0, r2)
+    x = r4
+    r5 = f2(x)
+    CPyList_SetItemUnsafe(r1, r2, r5)
+L3:
+    r6 = r2 + 1
+    r2 = r6
+    goto L1
+L4:
+    a = r1
+    return 1
+
 [case testListBuiltFromFinalStr]
 from typing import Final
 
@@ -729,6 +770,66 @@ L8:
     a = r2
     return 1
 
+[case testListBuiltFromBytesExpr_64bit]
+def f2(val: int) -> int:
+    return val + 2
+
+def test() -> None:
+    a = [f2(x) for x in b"abc"]
+
+[out]
+def f2(val):
+    val, r0 :: int
+L0:
+    r0 = CPyTagged_Add(val, 4)
+    return r0
+def test():
+    r0 :: bytes
+    r1 :: list
+    r2 :: native_int
+    r3, r4, r5 :: bit
+    r6, r7, r8, r9 :: int
+    r10 :: object
+    r11, x, r12 :: int
+    r13 :: object
+    r14 :: native_int
+    a :: list
+L0:
+    r0 = b'abc'
+    r1 = PyList_New(3)
+    r2 = 0
+L1:
+    r3 = r2 < 3 :: signed
+    if r3 goto L2 else goto L8 :: bool
+L2:
+    r4 = r2 <= 4611686018427387903 :: signed
+    if r4 goto L3 else goto L4 :: bool
+L3:
+    r5 = r2 >= -4611686018427387904 :: signed
+    if r5 goto L5 else goto L4 :: bool
+L4:
+    r6 = CPyTagged_FromInt64(r2)
+    r7 = r6
+    goto L6
+L5:
+    r8 = r2 << 1
+    r7 = r8
+L6:
+    r9 = CPyBytes_GetItem(r0, r7)
+    r10 = box(int, r9)
+    r11 = unbox(int, r10)
+    x = r11
+    r12 = f2(x)
+    r13 = box(int, r12)
+    CPyList_SetItemUnsafe(r1, r2, r13)
+L7:
+    r14 = r2 + 1
+    r2 = r14
+    goto L1
+L8:
+    a = r1
+    return 1
+
 [case testListBuiltFromFinalBytes_64bit]
 from typing import Final
 

mypyc/test-data/irbuild-tuple.test

@@ -319,6 +319,48 @@ L4:
     a = r3
     return 1
 
+[case testTupleBuiltFromStrExpr]
+def f2(val: str) -> str:
+    return val + "f2"
+
+def test() -> None:
+    a = tuple(f2(x) for x in "abc")
+
+[out]
+def f2(val):
+    val, r0, r1 :: str
+L0:
+    r0 = 'f2'
+    r1 = PyUnicode_Concat(val, r0)
+    return r1
+def test():
+    r0 :: str
+    r1 :: tuple
+    r2 :: native_int
+    r3 :: bit
+    r4, x, r5 :: str
+    r6 :: native_int
+    a :: tuple
+L0:
+    r0 = 'abc'
+    r1 = PyTuple_New(3)
+    r2 = 0
+L1:
+    r3 = r2 < 3 :: signed
+    if r3 goto L2 else goto L4 :: bool
+L2:
+    r4 = CPyStr_GetItemUnsafe(r0, r2)
+    x = r4
+    r5 = f2(x)
+    CPySequenceTuple_SetItemUnsafe(r1, r2, r5)
+L3:
+    r6 = r2 + 1
+    r2 = r6
+    goto L1
+L4:
+    a = r1
+    return 1
+
 [case testTupleBuiltFromFinalStr]
 from typing import Final
 
@@ -428,6 +470,66 @@ L8:
     a = r2
     return 1
 
+[case testTupleBuiltFromBytesExpr_64bit]
+def f2(val: int) -> int:
+    return val + 2
+
+def test() -> None:
+    a = tuple(f2(x) for x in b"abc")
+
+[out]
+def f2(val):
+    val, r0 :: int
+L0:
+    r0 = CPyTagged_Add(val, 4)
+    return r0
+def test():
+    r0 :: bytes
+    r1 :: tuple
+    r2 :: native_int
+    r3, r4, r5 :: bit
+    r6, r7, r8, r9 :: int
+    r10 :: object
+    r11, x, r12 :: int
+    r13 :: object
+    r14 :: native_int
+    a :: tuple
+L0:
+    r0 = b'abc'
+    r1 = PyTuple_New(3)
+    r2 = 0
+L1:
+    r3 = r2 < 3 :: signed
+    if r3 goto L2 else goto L8 :: bool
+L2:
+    r4 = r2 <= 4611686018427387903 :: signed
+    if r4 goto L3 else goto L4 :: bool
+L3:
+    r5 = r2 >= -4611686018427387904 :: signed
+    if r5 goto L5 else goto L4 :: bool
+L4:
+    r6 = CPyTagged_FromInt64(r2)
+    r7 = r6
+    goto L6
+L5:
+    r8 = r2 << 1
+    r7 = r8
+L6:
+    r9 = CPyBytes_GetItem(r0, r7)
+    r10 = box(int, r9)
+    r11 = unbox(int, r10)
+    x = r11
+    r12 = f2(x)
+    r13 = box(int, r12)
+    CPySequenceTuple_SetItemUnsafe(r1, r2, r13)
+L7:
+    r14 = r2 + 1
+    r2 = r14
+    goto L1
+L8:
+    a = r1
+    return 1
+
 [case testTupleBuiltFromFinalBytes_64bit]
 from typing import Final