commit ba5eb2de8cbd6fd2b033203b4c3c2d87e8b9f534
author David Tolnay <dtolnay@gmail.com> Fri Mar 06 09:58:20 2020 -0800
committer David Tolnay <dtolnay@gmail.com> Fri Mar 06 10:15:52 2020 -0800
tree fdcd3efbacb479b1281f7f13081effadac4eb6fa
parent 39d575fb04b235af187f786dbb04680fd77f9c9c

Add PartialEq impls for comparison against specific atoms

syntax/atom.rs

diff --git a/syntax/atom.rs b/syntax/atom.rs
index 57325d8..903f561 100644
--- a/syntax/atom.rs
+++ b/syntax/atom.rs
@@ -1,3 +1,4 @@
+use crate::syntax::Type;
 use proc_macro2::Ident;
 
 #[derive(Copy, Clone, PartialEq)]
@@ -38,3 +39,18 @@
         }
     }
 }
+
+impl PartialEq<Atom> for Ident {
+    fn eq(&self, atom: &Atom) -> bool {
+        Atom::from(self) == Some(*atom)
+    }
+}
+
+impl PartialEq<Atom> for Type {
+    fn eq(&self, atom: &Atom) -> bool {
+        match self {
+            Type::Ident(ident) => ident == atom,
+            _ => false,
+        }
+    }
+}