From 649c7f2fc4e71f4516bb7f5f6520d1961fb95cad Mon Sep 17 00:00:00 2001
From: Louis Pilfold <louis@lpil.uk>
Date: Thu, 12 Dec 2019 12:40:56 +0000
Subject: drop

---
 test/gleam/iterator_test.gleam | 36 ++++++++++++++++--------------------
 1 file changed, 16 insertions(+), 20 deletions(-)

(limited to 'test')

diff --git a/test/gleam/iterator_test.gleam b/test/gleam/iterator_test.gleam
index 079d57d..936621e 100644
--- a/test/gleam/iterator_test.gleam
+++ b/test/gleam/iterator_test.gleam
@@ -119,27 +119,23 @@ pub fn unfold_test() {
 }
 
 pub fn range_test() {
-  iterator.range(0, 0)
-  |> iterator.to_list
-  |> expect.equal(_, [])
-
-  iterator.range(1, 1)
-  |> iterator.to_list
-  |> expect.equal(_, [])
-
-  iterator.range(-1, -1)
-  |> iterator.to_list
-  |> expect.equal(_, [])
-
-  iterator.range(0, 1)
-  |> iterator.to_list
-  |> expect.equal(_, [0])
+  let test = fn(a, b, expected) {
+    iterator.range(a, b)
+    |> iterator.to_list
+    |> expect.equal(_, expected)
+  }
 
-  iterator.range(0, 5)
-  |> iterator.to_list
-  |> expect.equal(_, [0, 1, 2, 3, 4])
+  test(0, 0, [])
+  test(1, 1, [])
+  test(-1, -1, [])
+  test(0, 1, [0])
+  test(0, 5, [0, 1, 2, 3, 4])
+  test(1, -5, [1, 0, -1, -2, -3, -4])
+}
 
-  iterator.range(1, -5)
+pub fn drop_test() {
+  iterator.range(0, 10)
+  |> iterator.drop(_, 5)
   |> iterator.to_list
-  |> expect.equal(_, [1, 0, -1, -2, -3, -4])
+  |> expect.equal(_, [5, 6, 7, 8, 9])
 }
-- 
cgit v1.2.3