Add assertions on verify to zkprogram examples

Author: Shigoto-dev19

src/examples/zkprogram/hash-chain.ts

@@ -65,7 +65,7 @@ perfHashChain.end();
 perfHashChain.start('verify', 'chain');
 const isValid = await hashChain.verify(proof);
 perfHashChain.end();
-assert(isValid, 'Proof invalid');
+assert(isValid, 'Proof invalid!');
 
 // check that the output is correct
 let z = Array.from({ length: n }, () => 0).reduce((y) => Poseidon.hash([y]), x);

src/examples/zkprogram/mutual-recursion.ts

@@ -76,6 +76,7 @@ perfAdd.start('verify', 'performAddition');
 const validBaseCase = await verify(baseCase, addVk);
 perfAdd.end();
 console.log('ok?', validBaseCase);
+if (!validBaseCase) throw new Error('proof verification failed!');
 
 perfMultiply.start('prove', 'performMultiplication');
 const { proof: multiply1 } = await multiply.performMultiplication(Field(3), baseCase);
@@ -85,6 +86,7 @@ perfMultiply.start('verify', 'performMultiplication');
 const validMultiplication = await verify(multiply1, multiplyVk);
 perfMultiply.end();
 console.log('ok?', validMultiplication);
+if (!validMultiplication) throw new Error('proof verification failed!');
 
 console.log('Proving second (recursive) addition');
 const { proof: add2 } = await add.performAddition(
@@ -95,5 +97,6 @@ const { proof: add2 } = await add.performAddition(
 
 const validAddition = await verify(add2, addVk);
 console.log('ok?', validAddition);
+if (!validAddition) throw new Error('proof verification failed!');
 
 console.log('Result (should be 19):', add2.publicOutput.toBigInt());

src/examples/zkprogram/program-with-chunking.ts

@@ -37,7 +37,8 @@ let { proof } = await MyProgram.baseCase(Field(0));
 perf.end();
 
 perf.start('verify', 'baseCase');
-let ok = await MyProgram.verify(proof);
+let isValid = await MyProgram.verify(proof);
 perf.end();
 
-console.log('ok?', ok);
+console.log('isValid?', isValid);
+if (!isValid) throw new Error('proof verification failed!');

src/examples/zkprogram/program-with-input.ts

@@ -1,4 +1,4 @@
-import { SelfProof, Field, ZkProgram, verify, Proof, JsonProof, Provable } from 'o1js';
+import { Field, JsonProof, Proof, Provable, SelfProof, ZkProgram, verify } from 'o1js';
 
 let MyProgram = ZkProgram({
   name: 'example-with-input',
@@ -43,10 +43,12 @@ proof satisfies Proof<Field, void>;
 console.log('verify...');
 let ok = await verify(proof.toJSON(), verificationKey);
 console.log('ok?', ok);
+if (!ok) throw new Error('proof verification failed!');
 
 console.log('verify alternative...');
 ok = await MyProgram.verify(proof);
 console.log('ok (alternative)?', ok);
+if (!ok) throw new Error('proof verification failed!');
 
 console.log('proving step 1...');
 let { proof: proof1 } = await MyProgram.inductiveCase(1, proof);
@@ -55,19 +57,21 @@ proof1 = await testJsonRoundtrip(MyProgram.Proof, proof1);
 console.log('verify...');
 ok = await verify(proof1, verificationKey);
 console.log('ok?', ok);
+if (!ok) throw new Error('proof verification failed!');
 
 console.log('verify alternative...');
 ok = await MyProgram.verify(proof1);
 console.log('ok (alternative)?', ok);
+if (!ok) throw new Error('proof verification failed!');
 
 console.log('proving step 2...');
 let { proof: proof2 } = await MyProgram.inductiveCase(2, proof1);
 proof2 = await testJsonRoundtrip(MyProgram.Proof, proof2);
 
 console.log('verify...');
 ok = await verify(proof2.toJSON(), verificationKey);
-
 console.log('ok?', ok && proof2.publicInput.toString() === '2');
+if (!ok) throw new Error('proof verification failed!');
 
 function testJsonRoundtrip<
   P extends Proof<any, any>,

src/examples/zkprogram/program.ts

@@ -26,3 +26,4 @@ console.timeEnd('proving');
 console.log('verifying');
 let ok = await verify(result.proof, verificationKey);
 console.log('ok', ok);
+if (!ok) throw new Error('proof verification failed!');

src/examples/zkprogram/runtime-table/run.ts

@@ -27,5 +27,7 @@ let { proof } = await PayrollRuntimeTableZkProgram.verifyPayroll(
 perfPayroll.end();
 
 perfPayroll.start('verify', 'verifyPayroll');
-await verify(proof, verificationKey);
+let isValid = await verify(proof, verificationKey);
 perfPayroll.end();
+
+if (!isValid) throw new Error('proof verification failed!');

src/examples/zkprogram/side-loading/run.ts

@@ -59,7 +59,8 @@ perfMain.end();
 perfMain.start('verify', 'validateUsingTree');
 const validProof2 = await verify(proof2, mainVk);
 perfMain.end();
-console.log('ok?', validProof2);
+console.log('isValid2? ', validProof2);
+if (!validProof2) throw new Error('proof verification failed!');
 
 console.log('\nProving different method of child program');
 perfSide.start('prove', 'assertAndAdd');
@@ -76,4 +77,5 @@ const { proof: proof3 } = await mainProgram.validateUsingTree(
 );
 
 const validProof3 = await verify(proof3, mainVk);
-console.log('ok?', validProof3);
+console.log('isValid3? ', validProof3);
+if (!validProof3) throw new Error('proof verification failed!');

src/examples/zkprogram/small-big/run.ts

@@ -28,7 +28,8 @@ const { proof: proofBig } = await BigProgram.combinedHash(proofSmall.proof);
 perfBig.end();
 
 perfBig.start('verify', 'combinedHash');
-await verify(proofBig, verificationKeyBig);
+let isValid = await verify(proofBig, verificationKeyBig);
 perfBig.end();
 
+if (!isValid) throw new Error('proof verification failed!');
 console.log('Final Digest: ', proofBig.publicOutput.toHex());