extend test for traverse_path

Chia-Network · Feb 2, 2024 · 225269e · 225269e
1 parent b0e68a4
commit 225269e
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/traverse_path.rs b/src/traverse_path.rs
@@ -109,11 +109,17 @@ fn test_traverse_path() {
     let n1 = a.new_atom(&[0, 1, 2]).unwrap();
     let n2 = a.new_atom(&[4, 5, 6]).unwrap();
 
-    assert_eq!(traverse_path(&a, &[0], n1).unwrap(), Reduction(48, nul));
+    assert_eq!(traverse_path(&a, &[], n1).unwrap(), Reduction(44, nul));
     assert_eq!(traverse_path(&a, &[0b1], n1).unwrap(), Reduction(44, n1));
     assert_eq!(traverse_path(&a, &[0b1], n2).unwrap(), Reduction(44, n2));
 
     // cost for leading zeros
+    assert_eq!(traverse_path(&a, &[0], n1).unwrap(), Reduction(48, nul));
+    assert_eq!(traverse_path(&a, &[0, 0], n1).unwrap(), Reduction(52, nul));
+    assert_eq!(
+        traverse_path(&a, &[0, 0, 0], n1).unwrap(),
+        Reduction(54, nul)
+    );
     assert_eq!(
         traverse_path(&a, &[0, 0, 0, 0], n1).unwrap(),
         Reduction(60, nul)