privacy-pools-core/packages/circuits/tests/multiplier.test.ts

import { WitnessTester } from "circomkit";
import { circomkit } from "./common";

describe("multiplier", () => {
  const N = 3;
  let circuit: WitnessTester<["in"], ["out"]>;

  before(async () => {
    circuit = await circomkit.WitnessTester(`multiplier_${N}`, {
      file: "multiplier",
      template: "Multiplier",
      params: [N],
    });
  });

  it("should have correct number of constraints", async () => {
    await circuit.expectConstraintCount(N - 1);
  });

  it("should multiply correctly", async () => {
    const randomNumbers = Array.from({ length: N }, () => Math.floor(Math.random() * 100 * N));
    await circuit.expectPass({ in: randomNumbers }, { out: randomNumbers.reduce((prev, acc) => acc * prev) });
  });
});

describe("multiplier utilities", () => {
  describe("multiplication gate", () => {
    let circuit: WitnessTester<["in"], ["out"]>;

    before(async () => {
      circuit = await circomkit.WitnessTester("mulgate", {
        file: "multiplier",
        template: "MultiplicationGate",
        dir: "test/multiplier",
      });
    });

    it("should multiply correctly", async () => {
      await circuit.expectPass({ in: [7, 5] }, { out: 7 * 5 });
    });
  });
});