1const chai = require("chai");
 2const path = require("path");
 3const wasm_tester = require("circom_tester").wasm;
 4
 5const buildPoseidon = require("circomlibjs").buildPoseidon;
 6
 7const assert = chai.assert;
 8
 9describe("Poseidon Circuit test", function () {
10    let poseidon;
11    let F;
12    let circuit6;
13    let circuit3;
14    let circuitEx;
15
16    this.timeout(1000000);
17
18    before( async () => {
19        poseidon = await buildPoseidon();
20        F = poseidon.F;
21        circuit6 = await wasm_tester(path.join(__dirname, "circuits", "poseidon6_test.circom"));
22        circuit3 = await wasm_tester(path.join(__dirname, "circuits", "poseidon3_test.circom"));
23        circuitEx = await wasm_tester(path.join(__dirname, "circuits", "poseidonex_test.circom"));
24    });
25
26    it("Should check constrain of hash([1, 2]) t=6", async () => {
27        const w = await circuit6.calculateWitness({inputs: [1, 2, 0,0,0]}, true);
28
29        const res2 = poseidon([1,2,0,0,0]);
30
31        assert(F.eq(F.e("1018317224307729531995786483840663576608797660851238720571059489595066344487"), F.e(res2)));
32        await circuit6.assertOut(w, {out : F.toObject(res2)});
33        await circuit6.checkConstraints(w);
34    });
35
36    it("Should check constrain of hash([3, 4]) t=6", async () => {
37        const w = await circuit6.calculateWitness({inputs: [3, 4,5,10,23]});
38
39        const res2 = poseidon([3, 4,5,10,23]);
40
41        assert(F.eq(F.e("13034429309846638789535561449942021891039729847501137143363028890275222221409"), F.e(res2)));
42        await circuit6.assertOut(w, {out : F.toObject(res2)});
43        await circuit6.checkConstraints(w);
44    });
45
46
47    it("Should check constrain of hash([1, 2]) t=3", async () => {
48        const w = await circuit3.calculateWitness({inputs: [1, 2]});
49
50        const res2 = poseidon([1,2]);
51
52        assert(F.eq(F.e("7853200120776062878684798364095072458815029376092732009249414926327459813530"), F.e(res2)));
53        await circuit3.assertOut(w, {out : F.toObject(res2)});
54        await circuit3.checkConstraints(w);
55    });
56
57    it("Should check constrain of hash([3, 4]) t=3", async () => {
58        const w = await circuit3.calculateWitness({inputs: [3, 4]});
59
60        const res2 = poseidon([3, 4]);
61
62        assert(F.eq(F.e("14763215145315200506921711489642608356394854266165572616578112107564877678998"), F.e(res2)));
63        await circuit3.assertOut(w, {out : F.toObject(res2)});
64        await circuit3.checkConstraints(w);
65    });
66
67    it("Should check constrain of hash with state and 16 ins and outs", async () => {
68        const ins = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]
69        const w = await circuitEx.calculateWitness({inputs: ins, initialState: 17});
70
71        const res2 = poseidon(ins, 17, 17);
72        const res2f = [];
73        for (let i=0; i<res2.length; i++) {
74            res2f[i] = F.toObject(res2[i]);
75        }
76        await circuitEx.assertOut(w, {out : res2f});
77        await circuitEx.checkConstraints(w);
78    });
79
80});