Add tests and fix edge case

This commit is contained in:
Andrew Morris
2023-08-17 15:45:10 +10:00
parent 8313dcca3d
commit f59f68a861
2 changed files with 15 additions and 2 deletions

View File

@@ -115,8 +115,18 @@ mod tests {
tree.push(i);
}
tree.truncate(50);
assert_eq!(tree.len(), 100);
tree.truncate(100);
assert_eq!(tree.len(), 100);
tree.truncate(101);
assert_eq!(tree.len(), 100);
tree.truncate(1000);
assert_eq!(tree.len(), 100);
tree.truncate(50);
assert_eq!(tree.len(), 50);
assert_eq!(tree.depth(), 3);
@@ -125,7 +135,6 @@ mod tests {
}
tree.truncate(1);
assert_eq!(tree.len(), 1);
assert_eq!(tree.depth(), 1);
}

View File

@@ -193,6 +193,10 @@ impl<T: Clone, const N: usize> RadixTree<T, N> {
for p in path {
match tree.data_mut() {
RadixTreeData::Meta(meta) => {
if meta.len() <= p {
break;
}
meta.truncate(p + 1);
tree = &mut meta[p];
}