Scala : week2: implement filter

1 files changed, 15 insertions, 0 deletions

diff --git a/Scala/funsets/src/test/scala/funsets/FunSetSuite.scala b/Scala/funsets/src/test/scala/funsets/FunSetSuite.scala
index 88dac91..7c3fd53 100644
--- a/Scala/funsets/src/test/scala/funsets/FunSetSuite.scala
+++ b/Scala/funsets/src/test/scala/funsets/FunSetSuite.scala
@@ -162,4 +162,19 @@ class FunSetSuite extends FunSuite {
       assert(!contains(s, 3), "Diff none")
     }
   }
+
+  test("filter") {
+    new TestSets {
+      val s = filter(s1, (x => x == 1))
+      assert(!contains(s, 0), "Filter none")
+      assert(contains(s, 1), "Filter 1")
+      assert(!contains(s, 2), "Filter none")
+      assert(!contains(s, 3), "Filter none")
+      val v = filter(s1, (x => x == 0))
+      assert(!contains(v, 0), "Filter none")
+      assert(!contains(v, 1), "Filter none")
+      assert(!contains(v, 2), "Filter none")
+      assert(!contains(v, 3), "Filter none")
+    }
+  }
 }