From 915f43cc7695e937b90d2c8b3107bc66dfab5baa Mon Sep 17 00:00:00 2001
From: Louis Pilfold <louis@lpil.uk>
Date: Mon, 25 May 2020 11:27:06 +0100
Subject: set.take

---
 test/gleam/set_test.gleam | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

(limited to 'test')

diff --git a/test/gleam/set_test.gleam b/test/gleam/set_test.gleam
index b0f6ebd..efbdcfc 100644
--- a/test/gleam/set_test.gleam
+++ b/test/gleam/set_test.gleam
@@ -56,7 +56,7 @@ pub fn from_list_test() {
   |> set.from_list
   |> set.to_list
   |> list.sort(by: int.compare)
-  |> should.equal([1, 3, 3, 4])
+  |> should.equal([1, 2, 3, 4])
 }
 
 pub fn fold_test() {
@@ -64,3 +64,18 @@ pub fn fold_test() {
   |> set.from_list
   |> set.fold(from: 0, with: fn(m, a) { m + a })
 }
+
+pub fn filter_test() {
+  [1, 4, 6, 3, 675, 44, 67]
+  |> set.from_list()
+  |> set.filter(for: int.is_even)
+  |> set.to_list
+  |> should.equal([4, 6, 44])
+}
+
+pub fn take_test() {
+  [1, 2, 3]
+  |> set.from_list
+  |> set.take([1, 3, 5])
+  |> should.equal(set.from_list([1, 3]))
+}
-- 
cgit v1.2.3