circom/test/basiccases.js

const path = require("path");

const Scalar = require("ffjavascript").Scalar;
const F1Field = require("ffjavascript").F1Field;
const c_tester = require("../index.js").c_tester;
const wasm_tester = require("../index.js").wasm_tester;

const __P__ = Scalar.fromString("21888242871839275222246405745257275088548364400416034343698204186575808495617");

const Fr = new F1Field(__P__);

const basicCases = require("./basiccases.json");

function normalize(o) {
    if ((typeof(o) == "bigint") || o.isZero !== undefined)  {
        return Fr.e(o);
    } else if (Array.isArray(o)) {
        return o.map(normalize);
    } else if (typeof o == "object") {
        const res = {};
        for (let k in o) {
            res[k] = normalize(o[k]);
        }
        return res;
    } else {
        return Fr.e(o);
    }

}


async function doTest(tester, circuit, testVectors) {
    const cir = await tester(path.join(__dirname, "circuits", circuit));

    for (let i=0; i<testVectors.length; i++) {
        const w = await cir.calculateWitness(normalize(testVectors[i][0]));
//        console.log(testVectors[i][0]);
//        console.log(w);
//        console.log(testVectors[i][1]);
        await cir.assertOut(w, normalize(testVectors[i][1]) );
    }

    await cir.release();
}

describe("basic cases", function () {
    this.timeout(100000);

    for (let i=0; i< basicCases.length; i++) {
        it("c/c++ " + basicCases[i].name, async () => {
            await doTest(c_tester, basicCases[i].circuit, basicCases[i].tv);
        });
    }

    for (let i=0; i<basicCases.length; i++) {
        it("wasm " + basicCases[i].name, async () => {
            await doTest(wasm_tester, basicCases[i].circuit, basicCases[i].tv);
        });
    }

});