Use u32 index in test

rust-lang · Sep 24, 2021 · dfbf34b · dfbf34b
1 parent 72d96fa
commit dfbf34b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 9 deletions.
diff --git a/polonius-engine/src/output/mod.rs b/polonius-engine/src/output/mod.rs
@@ -618,16 +618,9 @@ fn compare_errors<Loan: Atom, Point: Atom>(
 mod tests {
     use super::*;
 
-    impl Atom for usize {
-        fn index(self) -> usize {
-            self
-        }
-    }
+    type Idx = u32;
 
-    fn compare(
-        errors1: &FxHashMap<usize, Vec<usize>>,
-        errors2: &FxHashMap<usize, Vec<usize>>,
-    ) -> bool {
+    fn compare(errors1: &FxHashMap<Idx, Vec<Idx>>, errors2: &FxHashMap<Idx, Vec<Idx>>) -> bool {
         let diff1 = compare_errors(errors1, errors2);
         let diff2 = compare_errors(errors2, errors1);
         assert_eq!(diff1, diff2);

diff --git a/polonius-facts/src/lib.rs b/polonius-facts/src/lib.rs
@@ -118,6 +118,12 @@ pub trait Atom: From<u32> + Into<u32> + Copy + Clone + Debug + Eq + Ord + Hash +
     fn index(self) -> usize;
 }
 
+impl Atom for u32 {
+    fn index(self) -> usize {
+        self as usize
+    }
+}
+
 pub trait FactTypes: Copy + Clone + Debug {
     type Origin: Atom;
     type Loan: Atom;