Scala : week2: implement intersect and diff

1 files changed, 24 insertions, 0 deletions

diff --git a/Scala/funsets/src/test/scala/funsets/FunSetSuite.scala b/Scala/funsets/src/test/scala/funsets/FunSetSuite.scala
index e61a56c..88dac91 100644
--- a/Scala/funsets/src/test/scala/funsets/FunSetSuite.scala
+++ b/Scala/funsets/src/test/scala/funsets/FunSetSuite.scala
@@ -138,4 +138,28 @@ class FunSetSuite extends FunSuite {
       assert(!contains(s, 3), "Union 3")
     }
   }
+
+  test("intersect") {
+    new TestSets {
+      val s = intersect(s1, s2)
+      assert(!contains(s, 1), "Intersect none")
+      assert(!contains(s, 2), "Intersect none")
+
+      val t = intersect(s1, s1)
+      assert(contains(t, 1), "Intersect 1")
+
+      val v = intersect(s2, s2)
+      assert(contains(v, 2), "Intersect 2")
+    }
+  }
+
+  test("diff") {
+    new TestSets {
+      val s = diff(s1, s2)
+      assert(!contains(s, 0), "Diff none")
+      assert(contains(s, 1), "Diff 1")
+      assert(!contains(s, 2), "Diff none")
+      assert(!contains(s, 3), "Diff none")
+    }
+  }
 }