Test for #18345

Closes #18345 [Cherry-picked 072b2e5]
scala · Dec 1, 2023 · f423521 · f423521
1 parent 4115f71
commit f423521
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/tests/pos/i18345.scala b/tests/pos/i18345.scala
@@ -0,0 +1,28 @@
+extension (vec: Seq[Int])
+  def iterate[T](body: (() => Int) => T): T =
+    val iterator = vec.iterator
+    body(() => iterator.nextOption().getOrElse(0))
+
+def withSequence[T](n: Int)(body: Seq[Int] => T): T =
+  body((0 to n))
+
+def test =
+
+  withSequence(2):
+    _.iterate: next =>
+      next() + next() + next() + next()
+
+  withSequence(2):
+    _.iterate:
+      next =>
+        next() + next() + next() + next()
+
+  withSequence(2): x =>
+    x.iterate:
+      next =>
+        next() + next() + next() + next()
+
+  withSequence(2): x =>
+    x.iterate: next =>
+      next() + next() + next() + next()
+