'return' for first class iterators

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/run/titer9.nim b/tests/run/titer9.nim
new file mode 100644
index 000000000..0d6c466c1
--- /dev/null
+++ b/tests/run/titer9.nim
@@ -0,0 +1,20 @@
+discard """
+  output: '''5
+14
+0'''
+"""
+
+iterator count(x: int, skip: bool): int {.closure.} =
+  if skip: return x+10
+  else: yield x+1
+
+  if skip: return x+10
+  else: yield x+2
+
+proc takeProc(x: iterator (x: int, skip: bool): int) =
+  echo x(4, false)
+  echo x(4, true)
+  echo x(4, false)
+
+takeProc(count)
+