feat(test): add mypyc tests for container creation from range

Author: BobTheBuidler

mypyc/test-data/irbuild-frozenset.test

@@ -113,3 +113,181 @@ L0:
     r5 = r4 >= 0 :: signed
     r6 = truncate r4: i32 to builtins.bool
     return r6
+
+[case testFrozenSetFromRange1]
+from typing import FrozenSet
+def fn() -> FrozenSet[int]:
+    return frozenset(range(3))
+[out]
+def fn():
+    r0, r1 :: object
+    r2 :: object[1]
+    r3 :: object_ptr
+    r4 :: object
+    r5 :: range
+    r6 :: frozenset
+L0:
+    r0 = load_address PyRange_Type
+    r1 = object 3
+    r2 = [r1]
+    r3 = load_address r2
+    r4 = PyObject_Vectorcall(r0, r3, 1, 0)
+    keep_alive r1
+    r5 = cast(range, r4)
+    r6 = PyFrozenSet_New(r5)
+    return r6
+
+[case testFrozenSetFromRange2]
+from typing import FrozenSet
+def fn() -> FrozenSet[int]:
+    return frozenset(range(1, 3))
+[out]
+def fn():
+    r0, r1, r2 :: object
+    r3 :: object[2]
+    r4 :: object_ptr
+    r5 :: object
+    r6 :: range
+    r7 :: frozenset
+L0:
+    r0 = load_address PyRange_Type
+    r1 = object 1
+    r2 = object 3
+    r3 = [r1, r2]
+    r4 = load_address r3
+    r5 = PyObject_Vectorcall(r0, r4, 2, 0)
+    keep_alive r1, r2
+    r6 = cast(range, r5)
+    r7 = PyFrozenSet_New(r6)
+    return r7
+
+[case testFrozenSetFromRange3]
+from typing import FrozenSet
+def fn() -> FrozenSet[int]:
+    return frozenset(range(1, 3, 2))
+[out]
+def fn():
+    r0, r1, r2, r3 :: object
+    r4 :: object[3]
+    r5 :: object_ptr
+    r6 :: object
+    r7 :: range
+    r8 :: frozenset
+L0:
+    r0 = load_address PyRange_Type
+    r1 = object 1
+    r2 = object 3
+    r3 = object 2
+    r4 = [r1, r2, r3]
+    r5 = load_address r4
+    r6 = PyObject_Vectorcall(r0, r5, 3, 0)
+    keep_alive r1, r2, r3
+    r7 = cast(range, r6)
+    r8 = PyFrozenSet_New(r7)
+    return r8
+
+[case testFrozenSetFromRange4]
+from typing import FrozenSet
+def fn() -> FrozenSet[str]:
+    return frozenset(str(i) for i in range(3))
+[out]
+def fn():
+    r0 :: list
+    r1 :: short_int
+    i :: int
+    r2 :: bit
+    r3 :: str
+    r4 :: i32
+    r5 :: bit
+    r6 :: short_int
+    r7 :: frozenset
+L0:
+    r0 = PyList_New(0)
+    r1 = 0
+    i = r1
+L1:
+    r2 = int_lt r1, 6
+    if r2 goto L2 else goto L4 :: bool
+L2:
+    r3 = CPyTagged_Str(i)
+    r4 = PyList_Append(r0, r3)
+    r5 = r4 >= 0 :: signed
+L3:
+    r6 = r1 + 2
+    r1 = r6
+    i = r6
+    goto L1
+L4:
+    r7 = PyFrozenSet_New(r0)
+    return r7
+
+[case testFrozenSetFromRange5]
+from typing import FrozenSet
+def fn() -> FrozenSet[str]:
+    return frozenset(str(i) for i in range(1, 3))
+[out]
+def fn():
+    r0 :: list
+    r1 :: short_int
+    i :: int
+    r2 :: bit
+    r3 :: str
+    r4 :: i32
+    r5 :: bit
+    r6 :: short_int
+    r7 :: frozenset
+L0:
+    r0 = PyList_New(0)
+    r1 = 2
+    i = r1
+L1:
+    r2 = int_lt r1, 6
+    if r2 goto L2 else goto L4 :: bool
+L2:
+    r3 = CPyTagged_Str(i)
+    r4 = PyList_Append(r0, r3)
+    r5 = r4 >= 0 :: signed
+L3:
+    r6 = r1 + 2
+    r1 = r6
+    i = r6
+    goto L1
+L4:
+    r7 = PyFrozenSet_New(r0)
+    return r7
+
+[case testFrozenSetFromRange6]
+from typing import FrozenSet
+def fn() -> FrozenSet[str]:
+    return frozenset(str(i) for i in range(1, 3, 2))
+[out]
+def fn():
+    r0 :: list
+    r1 :: short_int
+    i :: int
+    r2 :: bit
+    r3 :: str
+    r4 :: i32
+    r5 :: bit
+    r6 :: short_int
+    r7 :: frozenset
+L0:
+    r0 = PyList_New(0)
+    r1 = 2
+    i = r1
+L1:
+    r2 = int_lt r1, 6
+    if r2 goto L2 else goto L4 :: bool
+L2:
+    r3 = CPyTagged_Str(i)
+    r4 = PyList_Append(r0, r3)
+    r5 = r4 >= 0 :: signed
+L3:
+    r6 = r1 + 4
+    r1 = r6
+    i = r6
+    goto L1
+L4:
+    r7 = PyFrozenSet_New(r0)
+    return r7
+

mypyc/test-data/irbuild-lists.test

@@ -572,3 +572,181 @@ def sort_iterable(a):
 L0:
     r0 = CPySequence_Sort(a)
     return 1
+
+[case testListFromRange1]
+def fn() -> list[int]:
+    return list(range(3))
+[out]
+def fn():
+    r0, r1 :: object
+    r2 :: object[1]
+    r3 :: object_ptr
+    r4 :: object
+    r5 :: range
+    r6 :: list
+L0:
+    r0 = load_address PyRange_Type
+    r1 = object 3
+    r2 = [r1]
+    r3 = load_address r2
+    r4 = PyObject_Vectorcall(r0, r3, 1, 0)
+    keep_alive r1
+    r5 = cast(range, r4)
+    r6 = PySequence_List(r5)
+    return r6
+
+[case testListFromRange2]
+def fn() -> list[int]:
+    return list(range(1, 3))
+[out]
+def fn():
+    r0, r1, r2 :: object
+    r3 :: object[2]
+    r4 :: object_ptr
+    r5 :: object
+    r6 :: range
+    r7 :: list
+L0:
+    r0 = load_address PyRange_Type
+    r1 = object 1
+    r2 = object 3
+    r3 = [r1, r2]
+    r4 = load_address r3
+    r5 = PyObject_Vectorcall(r0, r4, 2, 0)
+    keep_alive r1, r2
+    r6 = cast(range, r5)
+    r7 = PySequence_List(r6)
+    return r7
+
+[case testListFromRange3]
+def fn() -> list[int]:
+    return list(range(1, 3, 2))
+[out]
+def fn():
+    r0, r1, r2, r3 :: object
+    r4 :: object[3]
+    r5 :: object_ptr
+    r6 :: object
+    r7 :: range
+    r8 :: list
+L0:
+    r0 = load_address PyRange_Type
+    r1 = object 1
+    r2 = object 3
+    r3 = object 2
+    r4 = [r1, r2, r3]
+    r5 = load_address r4
+    r6 = PyObject_Vectorcall(r0, r5, 3, 0)
+    keep_alive r1, r2, r3
+    r7 = cast(range, r6)
+    r8 = PySequence_List(r7)
+    return r8
+
+[case testListFromRange4]
+def fn() -> list[str]:
+    return list(str(i) for i in range(3))
+[out]
+def fn():
+    r0 :: list
+    r1 :: short_int
+    i :: int
+    r2 :: bit
+    r3 :: str
+    r4 :: i32
+    r5 :: bit
+    r6 :: short_int
+    r7 :: object
+    r8 :: list
+L0:
+    r0 = PyList_New(0)
+    r1 = 0
+    i = r1
+L1:
+    r2 = int_lt r1, 6
+    if r2 goto L2 else goto L4 :: bool
+L2:
+    r3 = CPyTagged_Str(i)
+    r4 = PyList_Append(r0, r3)
+    r5 = r4 >= 0 :: signed
+L3:
+    r6 = r1 + 2
+    r1 = r6
+    i = r6
+    goto L1
+L4:
+    r7 = PyObject_GetIter(r0)
+    r8 = PySequence_List(r7)
+    return r8
+
+[case testListFromRange5]
+def fn() -> list[str]:
+    return list(str(i) for i in range(1, 3))
+[out]
+def fn():
+    r0 :: list
+    r1 :: short_int
+    i :: int
+    r2 :: bit
+    r3 :: str
+    r4 :: i32
+    r5 :: bit
+    r6 :: short_int
+    r7 :: object
+    r8 :: list
+L0:
+    r0 = PyList_New(0)
+    r1 = 2
+    i = r1
+L1:
+    r2 = int_lt r1, 6
+    if r2 goto L2 else goto L4 :: bool
+L2:
+    r3 = CPyTagged_Str(i)
+    r4 = PyList_Append(r0, r3)
+    r5 = r4 >= 0 :: signed
+L3:
+    r6 = r1 + 2
+    r1 = r6
+    i = r6
+    goto L1
+L4:
+    r7 = PyObject_GetIter(r0)
+    r8 = PySequence_List(r7)
+    return r8
+
+[case testListFromRange6]
+def fn() -> list[str]:
+    return list(str(i) for i in range(1, 3, 2))
+[out]
+def fn():
+    r0 :: list
+    r1 :: short_int
+    i :: int
+    r2 :: bit
+    r3 :: str
+    r4 :: i32
+    r5 :: bit
+    r6 :: short_int
+    r7 :: object
+    r8 :: list
+L0:
+    r0 = PyList_New(0)
+    r1 = 2
+    i = r1
+L1:
+    r2 = int_lt r1, 6
+    if r2 goto L2 else goto L4 :: bool
+L2:
+    r3 = CPyTagged_Str(i)
+    r4 = PyList_Append(r0, r3)
+    r5 = r4 >= 0 :: signed
+L3:
+    r6 = r1 + 4
+    r1 = r6
+    i = r6
+    goto L1
+L4:
+    r7 = PyObject_GetIter(r0)
+    r8 = PySequence_List(r7)
+    return r8
+