fix tests

Author: BobTheBuidler

mypyc/test-data/irbuild-frozenset.test

@@ -189,104 +189,159 @@ L0:
 [case testFrozenSetFromRange4]
 from typing import FrozenSet
 def fn() -> FrozenSet[str]:
-    return frozenset(str(i) for i in range(3))
+    abc = tuple(range(3))
+    return frozenset(str(i) for i in abc)
 [out]
 def fn():
-    r0 :: list
-    r1 :: short_int
-    i :: int
-    r2 :: bit
-    r3 :: str
-    r4 :: i32
-    r5 :: bit
-    r6 :: short_int
-    r7 :: frozenset
+    r0, r1 :: object
+    r2 :: object[1]
+    r3 :: object_ptr
+    r4 :: object
+    r5 :: range
+    r6, abc :: tuple
+    r7 :: native_int
+    r8 :: list
+    r9 :: native_int
+    r10 :: bit
+    r11 :: object
+    r12, i :: int
+    r13 :: str
+    r14 :: native_int
+    r15 :: frozenset
 L0:
-    r0 = PyList_New(0)
-    r1 = 0
-    i = r1
+    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_Tuple(r5)
+    abc = r6
+    r7 = var_object_size abc
+    r8 = PyList_New(r7)
+    r9 = 0
 L1:
-    r2 = int_lt r1, 6
-    if r2 goto L2 else goto L4 :: bool
+    r10 = r9 < r7 :: signed
+    if r10 goto L2 else goto L4 :: bool
 L2:
-    r3 = CPyTagged_Str(i)
-    r4 = PyList_Append(r0, r3)
-    r5 = r4 >= 0 :: signed
+    r11 = CPySequenceTuple_GetItemUnsafe(abc, r9)
+    r12 = unbox(int, r11)
+    i = r12
+    r13 = CPyTagged_Str(i)
+    CPyList_SetItemUnsafe(r8, r9, r13)
 L3:
-    r6 = r1 + 2
-    r1 = r6
-    i = r6
+    r14 = r9 + 1
+    r9 = r14
     goto L1
 L4:
-    r7 = PyFrozenSet_New(r0)
-    return r7
+    r15 = PyFrozenSet_New(r8)
+    return r15
 
 [case testFrozenSetFromRange5]
 from typing import FrozenSet
 def fn() -> FrozenSet[str]:
-    return frozenset(str(i) for i in range(1, 3))
+    abc = tuple(range(1, 3))
+    return frozenset(str(i) for i in abc)
 [out]
 def fn():
-    r0 :: list
-    r1 :: short_int
-    i :: int
-    r2 :: bit
-    r3 :: str
-    r4 :: i32
-    r5 :: bit
-    r6 :: short_int
-    r7 :: frozenset
+    r0, r1, r2 :: object
+    r3 :: object[2]
+    r4 :: object_ptr
+    r5 :: object
+    r6 :: range
+    r7, abc :: tuple
+    r8 :: native_int
+    r9 :: list
+    r10 :: native_int
+    r11 :: bit
+    r12 :: object
+    r13, i :: int
+    r14 :: str
+    r15 :: native_int
+    r16 :: frozenset
 L0:
-    r0 = PyList_New(0)
-    r1 = 2
-    i = r1
+    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_Tuple(r6)
+    abc = r7
+    r8 = var_object_size abc
+    r9 = PyList_New(r8)
+    r10 = 0
 L1:
-    r2 = int_lt r1, 6
-    if r2 goto L2 else goto L4 :: bool
+    r11 = r10 < r8 :: signed
+    if r11 goto L2 else goto L4 :: bool
 L2:
-    r3 = CPyTagged_Str(i)
-    r4 = PyList_Append(r0, r3)
-    r5 = r4 >= 0 :: signed
+    r12 = CPySequenceTuple_GetItemUnsafe(abc, r10)
+    r13 = unbox(int, r12)
+    i = r13
+    r14 = CPyTagged_Str(i)
+    CPyList_SetItemUnsafe(r9, r10, r14)
 L3:
-    r6 = r1 + 2
-    r1 = r6
-    i = r6
+    r15 = r10 + 1
+    r10 = r15
     goto L1
 L4:
-    r7 = PyFrozenSet_New(r0)
-    return r7
+    r16 = PyFrozenSet_New(r9)
+    return r16
 
 [case testFrozenSetFromRange6]
 from typing import FrozenSet
 def fn() -> FrozenSet[str]:
-    return frozenset(str(i) for i in range(1, 3, 2))
+    abc = tuple(range(1, 3, 2))
+    return frozenset(str(i) for i in abc)
 [out]
 def fn():
-    r0 :: list
-    r1 :: short_int
-    i :: int
-    r2 :: bit
-    r3 :: str
-    r4 :: i32
-    r5 :: bit
-    r6 :: short_int
-    r7 :: frozenset
+    r0, r1, r2, r3 :: object
+    r4 :: object[3]
+    r5 :: object_ptr
+    r6 :: object
+    r7 :: range
+    r8, abc :: tuple
+    r9 :: native_int
+    r10 :: list
+    r11 :: native_int
+    r12 :: bit
+    r13 :: object
+    r14, i :: int
+    r15 :: str
+    r16 :: native_int
+    r17 :: frozenset
 L0:
-    r0 = PyList_New(0)
-    r1 = 2
-    i = r1
+    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_Tuple(r7)
+    abc = r8
+    r9 = var_object_size abc
+    r10 = PyList_New(r9)
+    r11 = 0
 L1:
-    r2 = int_lt r1, 6
-    if r2 goto L2 else goto L4 :: bool
+    r12 = r11 < r9 :: signed
+    if r12 goto L2 else goto L4 :: bool
 L2:
-    r3 = CPyTagged_Str(i)
-    r4 = PyList_Append(r0, r3)
-    r5 = r4 >= 0 :: signed
+    r13 = CPySequenceTuple_GetItemUnsafe(abc, r11)
+    r14 = unbox(int, r13)
+    i = r14
+    r15 = CPyTagged_Str(i)
+    CPyList_SetItemUnsafe(r10, r11, r15)
 L3:
-    r6 = r1 + 4
-    r1 = r6
-    i = r6
+    r16 = r11 + 1
+    r11 = r16
     goto L1
 L4:
-    r7 = PyFrozenSet_New(r0)
-    return r7
+    r17 = PyFrozenSet_New(r10)
+    return r17
+