From ae5730686b046cba6365989fc0f88b3b2644beee Mon Sep 17 00:00:00 2001 From: Louis Pilfold Date: Wed, 11 Dec 2019 23:35:59 +0000 Subject: repeat test --- src/gleam/iterator.gleam | 2 -- 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]) +} -- cgit v1.2.3