Merge branch 'nonfuzzy-fill' of github.com:ajna-finance/contracts int…

…o nonfuzzy-fill
ajna-finance · Jan 8, 2023 · 9b6e560 · 9b6e560
2 parents a0b12b7 + 7157a9d
commit 9b6e560
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/forge/utils/FenwickTreeInstance.sol b/tests/forge/utils/FenwickTreeInstance.sol
@@ -119,6 +119,7 @@ contract FenwickTreeInstance is DSTestPlus {
     ) external {
         uint256 i;
         uint256 amount;
+        uint256 cumulativeAmount;
 
         // Initialize and print seed for randomness
         setRandomSeed(seed_);
@@ -133,15 +134,16 @@ contract FenwickTreeInstance is DSTestPlus {
 
             // Update values
             add(i, amount);
-            amount_  -=  amount;
-            insertions_      -=  1;
+            amount_          -= amount;
+            insertions_      -= 1;
+            cumulativeAmount += amount;
 
             // Verify tree sum
-            assertEq(deposits.treeSum(), amount_ - amount_);
+            assertEq(deposits.treeSum(), cumulativeAmount);
 
             if (trackInserts)  inserts.push(i);
         }
 
-        assertEq(deposits.treeSum(), amount_);
+        assertEq(deposits.treeSum(), cumulativeAmount);
     }
 }