diff options
| -rw-r--r-- | src/gleam/iterator.gleam | 2 | ||||
| -rw-r--r-- | test/gleam/iterator_test.gleam | 7 |
2 files changed, 7 insertions, 2 deletions
diff --git a/src/gleam/iterator.gleam b/src/gleam/iterator.gleam index 32bed9c..a2ca7d8 100644 --- a/src/gleam/iterator.gleam +++ b/src/gleam/iterator.gleam @@ -59,7 +59,6 @@ pub fn repeatedly(f: fn() -> element) -> Iterator(element) { unfold(Nil, fn(acc) { Next(f(), acc) }) } -// TODO: test // TODO: document pub fn repeat(x: element) -> Iterator(element) { repeatedly(fn() { x }) @@ -154,7 +153,6 @@ fn do_filter(iterator, predicate) { } } -// TODO: test // TODO: document pub fn filter(iterator: Iterator(a), for predicate: fn(a) -> Bool) -> Iterator(a) { iterator diff --git a/test/gleam/iterator_test.gleam b/test/gleam/iterator_test.gleam index f072889..b587d12 100644 --- a/test/gleam/iterator_test.gleam +++ b/test/gleam/iterator_test.gleam @@ -92,3 +92,10 @@ pub fn filter_test() { test([1, 2, 3, 4, 5], even) test([1, 2, 3, 4, 5, 6], even) } + +pub fn repeat_test() { + 1 + |> iterator.repeat + |> iterator.take(_, 5) + |> expect.equal(_, [1, 1, 1, 1, 1]) +} |