add regression unit test

o1-labs · Apr 10, 2024 · fe30b9a · fe30b9a
1 parent cbf4975
commit fe30b9a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/lib/provable/test/merkle-tree.unit-test.ts b/src/lib/provable/test/merkle-tree.unit-test.ts
@@ -2,6 +2,7 @@ import { Bool, Field } from '../wrapped.js';
 import { maybeSwap } from '../merkle-tree.js';
 import { Random, test } from '../../testing/property.js';
 import { expect } from 'expect';
+import { MerkleMap } from '../merkle-map.js';
 
 test(Random.bool, Random.field, Random.field, (b, x, y) => {
   let [x0, y0] = maybeSwap(Bool(!!b), Field(x), Field(y));
@@ -15,3 +16,10 @@ test(Random.bool, Random.field, Random.field, (b, x, y) => {
     expect(y0.toBigInt()).toEqual(x);
   }
 });
+
+test(Random.field, (key) => {
+  let map = new MerkleMap();
+  let witness = map.getWitness(Field(key));
+  let [, calculatedKey] = witness.computeRootAndKey(Field(0));
+  expect(calculatedKey.toBigInt()).toEqual(key);
+});