From 70295da0ba0a5b6ab347d3fef7d3dfa956510694 Mon Sep 17 00:00:00 2001 From: Michael Bradley Date: Fri, 31 Jan 2025 01:59:43 -0500 Subject: [PATCH] Check more return values are as expected By "Check" I mean I'm just unwrapping ones that should exist because I'm lazy --- src/backing/indexed/binary_heap.rs | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/src/backing/indexed/binary_heap.rs b/src/backing/indexed/binary_heap.rs index 05c73e6..2f5dbc6 100644 --- a/src/backing/indexed/binary_heap.rs +++ b/src/backing/indexed/binary_heap.rs @@ -78,12 +78,14 @@ impl if smaller_child < self.data[i] { // Swap parent with child self.data[smaller_child_index] = self.data[i].clone(); - self.indices.insert( - self.data[smaller_child_index].clone().data(), - smaller_child_index, - ); + self.indices + .insert( + self.data[smaller_child_index].clone().data(), + smaller_child_index, + ) + .unwrap(); self.data[i] = smaller_child.clone(); - self.indices.insert(smaller_child.data(), i); + self.indices.insert(smaller_child.data(), i).unwrap(); // Repeat process with child self.sift_down(smaller_child_index) @@ -113,7 +115,7 @@ impl } else { self.data.pop().unwrap(); } - self.indices.remove(pair.get_data()); + self.indices.remove(pair.get_data()).unwrap(); Some(pair) } } @@ -162,7 +164,9 @@ impl In } } else { let final_index = self.data.len(); - self.indices.insert(data.clone(), final_index); + if let Some(_) = self.indices.insert(data.clone(), final_index) { + panic!("Item was not consistently hashed") + } self.data.push(Pair::new(data, priority)); self.sift_up(final_index).unwrap(); }