From 21e774dea2d678b90d5eb872eb9f0d9924c818ab Mon Sep 17 00:00:00 2001
From: Louis Pilfold <louis@lpil.uk>
Date: Wed, 11 Dec 2019 07:52:17 +0000
Subject: take

---
 test/gleam/iterator_test.gleam | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'test')

diff --git a/test/gleam/iterator_test.gleam b/test/gleam/iterator_test.gleam
index fbbf1c6..2d9ee3f 100644
--- a/test/gleam/iterator_test.gleam
+++ b/test/gleam/iterator_test.gleam
@@ -1,6 +1,10 @@
 import gleam/expect
 import gleam/iterator
+import gleam/list
 
+// TODO: Property tests
+
+// a |> from_list |> to_list == a
 pub fn to_from_list_test() {
   let test = fn(subject) {
     subject
@@ -14,3 +18,24 @@ pub fn to_from_list_test() {
   test([1, 2])
   test([1, 2, 4, 8])
 }
+
+// a |> from_list |> take(_, n) == a |> list.take(_, n)
+pub fn take_test() {
+  let test = fn(n, subject) {
+    subject
+    |> iterator.from_list
+    |> iterator.take(_, n)
+    |> expect.equal(_, list.take(subject, n))
+  }
+
+  test(0, [])
+  test(1, [])
+  test(-1, [])
+  test(0, [0])
+  test(1, [0])
+  test(-1, [0])
+  test(0, [0, 1, 2, 3, 4])
+  test(1, [0, 1, 2, 3, 4])
+  test(2, [0, 1, 2, 3, 4])
+  test(22, [0, 1, 2, 3, 4])
+}
-- 
cgit v1.2.3