From 264035e99f0f6e123f9f9172714f03212b621ddf Mon Sep 17 00:00:00 2001
From: lovesh <lovesh.bond@gmail.com>
Date: Fri, 29 Mar 2024 01:37:41 +0530
Subject: [PATCH] Update test

Signed-off-by: lovesh <lovesh.bond@gmail.com>
---
 tests/js/kb-universal-accumulator.spec.ts | 35 +++++++++++------------
 1 file changed, 17 insertions(+), 18 deletions(-)

diff --git a/tests/js/kb-universal-accumulator.spec.ts b/tests/js/kb-universal-accumulator.spec.ts
index 5b9ee17..41b8b4c 100644
--- a/tests/js/kb-universal-accumulator.spec.ts
+++ b/tests/js/kb-universal-accumulator.spec.ts
@@ -804,6 +804,14 @@ describe("For KB universal accumulator witness update", () => {
             [],
             sk
         );
+        let [memPublicInfo_, nonMemPublicInfo_] = publicInfoForBothKBUniversalWitnessUpdate(
+            accumulator,
+            addBatch0,
+            [],
+            sk
+        );
+        expect(areUint8ArraysEqual(memPublicInfo0, memPublicInfo_)).toEqual(true);
+        expect(areUint8ArraysEqual(nonMemPublicInfo0, nonMemPublicInfo_)).toEqual(true);
 
         accumulator = kbUniversalAccumulatorAddBatch(
             accumulator,
@@ -845,15 +853,6 @@ describe("For KB universal accumulator witness update", () => {
             )
         ).toBe(true);
 
-        // let [memPublicInfo_, nonMemPublicInfo_] = publicInfoForBothKBUniversalWitnessUpdate(
-        //     accumulator,
-        //     addBatch0,
-        //     [],
-        //     sk
-        // );
-        // expect(areUint8ArraysEqual(memPublicInfo0, memPublicInfo_)).toEqual(true);
-        // expect(areUint8ArraysEqual(nonMemPublicInfo0, nonMemPublicInfo_)).toEqual(true);
-
         const addBatch1 = [
             domain[3],
             domain[4],
@@ -873,6 +872,15 @@ describe("For KB universal accumulator witness update", () => {
             sk
         );
 
+        [memPublicInfo_, nonMemPublicInfo_] = publicInfoForBothKBUniversalWitnessUpdate(
+            accumulator,
+            addBatch1,
+            remBatch1,
+            sk
+        );
+        expect(areUint8ArraysEqual(memPublicInfo1, memPublicInfo_)).toEqual(true);
+        expect(areUint8ArraysEqual(nonMemPublicInfo1, nonMemPublicInfo_)).toEqual(true);
+
         accumulator = kbUniversalAccumulatorBatchUpdates(
             accumulator,
             addBatch1,
@@ -914,15 +922,6 @@ describe("For KB universal accumulator witness update", () => {
             )
         ).toBe(true);
 
-        // [memPublicInfo_, nonMemPublicInfo_] = publicInfoForBothKBUniversalWitnessUpdate(
-        //     accumulator,
-        //     addBatch1,
-        //     remBatch1,
-        //     sk
-        // );
-        // expect(areUint8ArraysEqual(memPublicInfo1, memPublicInfo_)).toEqual(true);
-        // expect(areUint8ArraysEqual(nonMemPublicInfo1, nonMemPublicInfo_)).toEqual(true);
-
         const addBatch2 = [
             domain[5],
             domain[6],