Add Equals method

basp1 · Jul 10, 2019 · 0105fad · 0105fad
1 parent c954c8a
commit 0105fad
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/FlatMap.cs b/src/FlatMap.cs
@@ -1,6 +1,6 @@
 using System;
 using System.Collections.Generic;
-using System.Text;
+using System.Linq;
 
 namespace sharpcode
 {
@@ -111,6 +111,16 @@ public Value this[Key key]
             }
         }
 
+        public override bool Equals(object obj)
+        {
+            return Equals(obj as FlatMap<Key, Value>);
+        }
+
+        public bool Equals(FlatMap<Key, Value> that)
+        {
+            return keys.SequenceEqual(that.keys) && values.SequenceEqual(that.values);
+        }
+
         bool Find(Key key, ref Value value)
         {
             int index = LowerBound(key);

diff --git a/src/SegmentTree.cs b/src/SegmentTree.cs
@@ -49,6 +49,11 @@ public int Capacity
             }
         }
 
+        public override bool Equals(object obj)
+        {
+            return Equals(obj as SegmentTree<T>);
+        }
+
         public bool Equals(SegmentTree<T> that)
         {
             if (capacity != that.Capacity)