You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

75 lines
2.7 KiB

5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
  1. const chai = require("chai");
  2. const path = require("path");
  3. var blake2b = require("blake2b");
  4. const tester = require("circom").tester;
  5. const poseidon = require("../src/poseidon.js");
  6. const assert = chai.assert;
  7. describe("Blake2b version test", function() {
  8. it("Should give the expected output for blake2b version", async () => {
  9. var output = new Uint8Array(32);
  10. var input = Buffer.from("poseidon_constants");
  11. const h = blake2b(output.length).update(input).digest("hex");
  12. assert.equal("e57ba154fb2c47811dc1a2369b27e25a44915b4e4ece4eb8ec74850cb78e01b1", h);
  13. });
  14. });
  15. describe("Poseidon Circuit test", function () {
  16. let circuit6;
  17. let circuit3;
  18. this.timeout(100000);
  19. before( async () => {
  20. circuit6 = await tester(path.join(__dirname, "circuits", "poseidon6_test.circom"));
  21. circuit3 = await tester(path.join(__dirname, "circuits", "poseidon3_test.circom"));
  22. });
  23. it("Should check constrain of hash([1, 2]) t=6", async () => {
  24. const w = await circuit6.calculateWitness({inputs: [1, 2]}, true);
  25. const hash = poseidon.createHash(6, 8, 57);
  26. const res2 = hash([1,2]);
  27. assert.equal("12242166908188651009877250812424843524687801523336557272219921456462821518061", res2.toString());
  28. await circuit6.assertOut(w, {out : res2});
  29. await circuit6.checkConstraints(w);
  30. });
  31. it("Should check constrain of hash([3, 4]) t=6", async () => {
  32. const w = await circuit6.calculateWitness({inputs: [3, 4]});
  33. const hash = poseidon.createHash(6, 8, 57);
  34. const res2 = hash([3, 4]);
  35. assert.equal("17185195740979599334254027721507328033796809509313949281114643312710535000993", res2.toString());
  36. await circuit6.assertOut(w, {out : res2});
  37. await circuit6.checkConstraints(w);
  38. });
  39. it("Should check constrain of hash([1, 2]) t=3", async () => {
  40. const w = await circuit3.calculateWitness({inputs: [1, 2]});
  41. const hash = poseidon.createHash(3, 8, 57);
  42. const res2 = hash([1,2]);
  43. assert.equal("2104035019328376391822106787753454168168617545136592089411833517434990977743", res2.toString());
  44. await circuit3.assertOut(w, {out : res2});
  45. await circuit3.checkConstraints(w);
  46. });
  47. it("Should check constrain of hash([3, 4]) t=3", async () => {
  48. const w = await circuit3.calculateWitness({inputs: [3, 4]});
  49. const hash = poseidon.createHash(3, 8, 57);
  50. const res2 = hash([3, 4]);
  51. assert.equal("12456141564250880945411182508630957604732712316993112736876413121277158512223", res2.toString());
  52. await circuit3.assertOut(w, {out : res2});
  53. await circuit3.checkConstraints(w);
  54. });
  55. });