From 3bbd4a1bf2d66d75a5547dcc15049174857850f6 Mon Sep 17 00:00:00 2001 From: Louis Pilfold Date: Thu, 12 Dec 2019 12:06:28 +0000 Subject: unfold test --- test/gleam/iterator_test.gleam | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'test') diff --git a/test/gleam/iterator_test.gleam b/test/gleam/iterator_test.gleam index 447e912..fe0a854 100644 --- a/test/gleam/iterator_test.gleam +++ b/test/gleam/iterator_test.gleam @@ -107,3 +107,13 @@ pub fn cycle_test() { |> iterator.take(_, 9) |> expect.equal(_, [1, 2, 3, 1, 2, 3, 1, 2, 3]) } + +pub fn unfold_test() { + iterator.unfold(2, fn(acc) { iterator.Next(acc, acc * 2) }) + |> iterator.take(_, 5) + |> expect.equal(_, [2, 4, 8, 16, 32]) + + iterator.unfold(2, fn(_) { iterator.Done }) + |> iterator.take(_, 5) + |> expect.equal(_, []) +} -- cgit v1.2.3