Merge pull request xacrimon#332 from noahbkim/eq

Implement Eq and PartialEq for DashMap and DashSet

Author: xacrimon

src/lib.rs

@@ -1344,6 +1344,21 @@ where
     }
 }
 
+impl<'a, K: 'a + Eq + Hash, V: 'a + PartialEq, S: BuildHasher + Clone> PartialEq
+    for DashMap<K, V, S>
+{
+    fn eq(&self, other: &Self) -> bool {
+        self.len() == other.len()
+            && self.iter().all(|r| {
+                other
+                    .get(r.key())
+                    .map_or(false, |ro| r.value() == ro.value())
+            })
+    }
+}
+
+impl<'a, K: 'a + Eq + Hash, V: 'a + Eq, S: BuildHasher + Clone> Eq for DashMap<K, V, S> {}
+
 impl<K: Eq + Hash, V, S: BuildHasher + Clone> IntoIterator for DashMap<K, V, S> {
     type Item = (K, V);
 
@@ -1432,6 +1447,25 @@ mod tests {
         assert_eq!(dm.get(&0).unwrap().value(), &0);
     }
 
+    #[test]
+    fn test_equal() {
+        let dm1 = DashMap::new();
+        let dm2 = DashMap::new();
+        assert_eq!(dm1, dm2);
+
+        dm1.insert(0, "Hello, world!");
+        assert_ne!(dm1, dm2);
+
+        dm1.insert(1, "Goodbye, world!");
+        assert_ne!(dm1, dm2);
+
+        dm2.insert(0, "Hello, world!");
+        assert_ne!(dm1, dm2);
+
+        dm2.insert(1, "Goodbye, world!");
+        assert_eq!(dm1, dm2);
+    }
+
     #[test]
     fn test_multiple_hashes() {
         let dm: DashMap<u32, u32> = DashMap::default();

src/set.rs

@@ -383,6 +383,14 @@ impl<'a, K: 'a + Eq + Hash, S: BuildHasher + Clone> DashSet<K, S> {
     }
 }
 
+impl<K: Eq + Hash, S: BuildHasher + Clone> PartialEq for DashSet<K, S> {
+    fn eq(&self, other: &Self) -> bool {
+        self.len() == other.len() && self.iter().all(|r| other.contains(r.key()))
+    }
+}
+
+impl<K: Eq + Hash, S: BuildHasher + Clone> Eq for DashSet<K, S> {}
+
 impl<K: Eq + Hash, S: BuildHasher + Clone> IntoIterator for DashSet<K, S> {
     type Item = K;
 
@@ -450,6 +458,25 @@ mod tests {
         assert_eq!(set.get(&0).as_deref(), Some(&0));
     }
 
+    #[test]
+    fn test_equal() {
+        let set1 = DashSet::new();
+        let set2 = DashSet::new();
+        assert_eq!(set1, set2);
+
+        set1.insert("Hello, world!");
+        assert_ne!(set1, set2);
+
+        set1.insert("Goodbye, world!");
+        assert_ne!(set1, set2);
+
+        set2.insert("Hello, world!");
+        assert_ne!(set1, set2);
+
+        set2.insert("Goodbye, world!");
+        assert_eq!(set1, set2);
+    }
+
     #[test]
     fn test_multiple_hashes() {
         let set = DashSet::<u32>::default();