'var T' for iterators

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/accept/run/tmoditer.nim b/tests/accept/run/tmoditer.nim
new file mode 100644
index 000000000..1e6be37e4
--- /dev/null
+++ b/tests/accept/run/tmoditer.nim
@@ -0,0 +1,29 @@
+discard """
+  output: "XXXXX01234"
+"""
+
+iterator modPairs(a: var array[0..4,string]): tuple[key: int, val: var string] =
+  for i in 0..a.high:
+    yield (i, a[i])
+
+iterator modItems*[T](a: var array[0..4,T]): var T =
+  for i in 0..a.high:
+    yield a[i]
+
+var
+  arr = ["a", "b", "c", "d", "e"]
+
+for a in modItems(arr):
+  a = "X"
+
+for a in items(arr):
+  stdout.write(a)
+
+for i, a in modPairs(arr):
+  a = $i
+
+for a in items(arr):
+  stdout.write(a)
+
+echo ""
+