add splitBytesToWords unit test

This commit is contained in:
turnoffthiscomputer
2024-06-18 11:48:05 +02:00
parent fed5740745
commit 6e1b374925
2 changed files with 39 additions and 0 deletions

View File

@@ -0,0 +1,33 @@
import { wasm as wasm_tester } from "circom_tester";
import path from "path";
import { bigIntToChunkedBytes, Uint8ArrayToCharArray } from "@zk-email/helpers/src/binary-format";
describe("SplitBytesToWords Helper unit test", () => {
jest.setTimeout(0.1 * 60 * 1000);
let circuit: any;
beforeAll(async () => {
circuit = await wasm_tester(
path.join(__dirname, "./test-circuits/splitBytesToWords-test.circom"),
{
recompile: true,
include: path.join(__dirname, "../../../node_modules"),
// output: path.join(__dirname, "./compiled-test-circuits"),
}
);
});
it("should split correctly according to bigIntToChunkedBytes function", async function () {
const bytes = new Uint8Array(256).map(() => Math.floor(Math.random() * 256));
const bytesBigInt = bytes.reduce((acc, val) => (acc << 8n) | BigInt(val), 0n);
const ts_split_to_words = bigIntToChunkedBytes(bytesBigInt, 121, 17);
const ts_split_to_words_bigint = ts_split_to_words.map((word) => BigInt(word));
const witness = await circuit.calculateWitness({
in: Uint8ArrayToCharArray(bytes)
});
await circuit.assertOut(witness, { out: ts_split_to_words_bigint });
});
});

View File

@@ -0,0 +1,6 @@
pragma circom 2.1.6;
include "../../utils/bytes.circom";
component main = SplitBytesToWords(256,121,17);