Test try variants

garro95 · Aug 9, 2024 · 3530611 · 3530611
1 parent 56fe3fb
commit 3530611
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/tests/double_priority_queue.rs b/tests/double_priority_queue.rs
@@ -383,6 +383,28 @@ mod doublepq_tests {
         assert_eq!(queue.capacity(), 100);
     }
 
+    #[test]
+    fn try_reserve() {
+        use std::collections::hash_map::RandomState;
+        let mut queue = DoublePriorityQueue::<i32, i32, RandomState>::default();
+
+        queue.try_reserve(100).unwrap();
+
+        assert_eq!(queue.len(), 0);
+        assert!(queue.capacity() >= 100);
+    }
+
+    #[test]
+    fn try_reserve_exact() {
+        use std::collections::hash_map::RandomState;
+        let mut queue = DoublePriorityQueue::<i32, i32, RandomState>::default();
+
+        queue.try_reserve_exact(100).unwrap();
+
+        assert_eq!(queue.len(), 0);
+        assert_eq!(queue.capacity(), 100);
+    }
+
     #[test]
     fn extend() {
         let mut pq = DoublePriorityQueue::new();

diff --git a/tests/priority_queue.rs b/tests/priority_queue.rs
@@ -366,6 +366,28 @@ mod pqueue_tests {
         assert_eq!(queue.capacity(), 100);
     }
 
+    #[test]
+    fn try_reserve() {
+        use std::collections::hash_map::RandomState;
+        let mut queue = PriorityQueue::<i32, i32, RandomState>::default();
+
+        queue.try_reserve(100).unwrap();
+
+        assert_eq!(queue.len(), 0);
+        assert!(queue.capacity() >= 100);
+    }
+
+    #[test]
+    fn try_reserve_exact() {
+        use std::collections::hash_map::RandomState;
+        let mut queue = PriorityQueue::<i32, i32, RandomState>::default();
+
+        queue.try_reserve_exact(100).unwrap();
+
+        assert_eq!(queue.len(), 0);
+        assert_eq!(queue.capacity(), 100);
+    }
+
     #[test]
     fn extend() {
         let mut pq = PriorityQueue::new();