Update irbuild-tuple.test

Author: BobTheBuidler

mypyc/test-data/irbuild-tuple.test

@@ -748,13 +748,16 @@ def varint() -> int:
  # this helper lets us break constant folding for this test
  return 2
 
-source: Final = (1, varint(), 3)
+source_final: Final = (1, varint(), 3)
+source_var = (1, varint(), 3)
 
 def f(val: int) -> bool:
  return val % 2 == 0
 
-def test() -> None:
- a = tuple(f(x) for x in source)
+def test_final_can_fold() -> None:
+ a = tuple(f(x) for x in source_final)
+def test_var_can_not_fold() -> None:
+ a = tuple(f(x) for x in source_var)
 [out]
 def varint():
 L0:
@@ -766,7 +769,7 @@ L0:
  r0 = CPyTagged_Remainder(val, 4)
  r1 = int_eq r0, 0
  return r1
-def test():
+def test_final_can_fold():
  r0 :: tuple[int, int, int]
  r1 :: bool
  r2 :: object
@@ -785,10 +788,10 @@ def test():
  r17 :: native_int
  a :: tuple
 L0:
- r0 = __main__.source :: static
+ r0 = __main__.source_final :: static
  if is_error(r0) goto L1 else goto L2
 L1:
- r1 = raise NameError('value for final name "source" was not set')
+ r1 = raise NameError('value for final name "source_final" was not set')
  unreachable
 L2:
  r2 = box(tuple[int, int, int], r0)
@@ -818,6 +821,38 @@ L5:
 L6:
  a = r10
  return 1
+def test_var_can_not_fold():
+ source :: tuple
+ r0 :: native_int
+ r1 :: tuple
+ r2 :: native_int
+ r3 :: bit
+ r4 :: object
+ r5, x, r6 :: bool
+ r7 :: object
+ r8 :: native_int
+ a :: tuple
+L0:
+ r0 = var_object_size source
+ r1 = PyTuple_New(r0)
+ r2 = 0
+L1:
+ r3 = r2 < r0 :: signed
+ if r3 goto L2 else goto L4 :: bool
+L2:
+ r4 = CPySequenceTuple_GetItemUnsafe(source, r2)
+ r5 = unbox(bool, r4)
+ x = r5
+ r6 = f(x)
+ r7 = box(bool, r6)
+ CPySequenceTuple_SetItemUnsafe(r1, r2, r7)
+L3:
+ r8 = r2 + 1
+ r2 = r8
+ goto L1
+L4:
+ a = r1
+ return 1
 
 [case testTupleBuiltFromVariableLengthTuple]
 from typing import Tuple