feat: impl PartialOrd for newtypes to reference

Author: molpopgen

src/sys/macros.rs

@@ -134,6 +134,30 @@ macro_rules! impl_id_traits {
             }
         }
 
+        impl PartialEq<&$idtype> for $idtype {
+            fn eq(&self, other: &&$idtype) -> bool {
+                self.0 == other.0
+            }
+        }
+
+        impl PartialEq<$idtype> for &$idtype {
+            fn eq(&self, other: &$idtype) -> bool {
+                self.0 == other.0
+            }
+        }
+
+        impl PartialOrd<&$idtype> for $idtype {
+            fn partial_cmp(&self, other: &&$idtype) -> Option<std::cmp::Ordering> {
+                self.partial_cmp(&other.0)
+            }
+        }
+
+        impl PartialOrd<$idtype> for &$idtype {
+            fn partial_cmp(&self, other: &$idtype) -> Option<std::cmp::Ordering> {
+                self.0.partial_cmp(&other.0)
+            }
+        }
+
         impl Default for $idtype {
             fn default() -> Self {
                 Self::NULL

src/sys/newtypes.rs

@@ -243,3 +243,20 @@ fn test_try_from_reference() {
     let y = NodeId::from(2);
     assert_eq!(2, usize::try_from(&y).unwrap());
 }
+
+#[test]
+fn test_eq_to_self_ref() {
+    let x = NodeId::from(1);
+    let rx = &x;
+    assert_eq!(x, rx);
+    assert_eq!(rx, x);
+}
+
+#[test]
+fn test_ord_to_self_ref() {
+    let x = NodeId::from(1);
+    let y = NodeId::from(2);
+    let ry = &y;
+    assert!(x < ry);
+    assert!(ry > x);
+}

tests/book_trees.rs

@@ -86,7 +86,7 @@ fn initialize_from_table_collection() {
                 // The children function returns another iterator
                 let _siblings = tree
                     .children(parent)
-                    .filter(|child| child != &node)
+                    .filter(|child| child != node)
                     .collect::<Vec<_>>();
             }
         }