add invariant test

Uniswap · Apr 25, 2024 · bc5f8c0 · bc5f8c0
1 parent 4c7dafc
commit bc5f8c0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/libraries/UnsafeMath.t.sol b/test/libraries/UnsafeMath.t.sol
@@ -24,6 +24,12 @@ contract UnsafeMathTest is Test {
     }
 
     function test_fuzz_divRoundingUp(uint256 x, uint256 y) public {
+        vm.assume(y != 0);
+        uint256 result = x.divRoundingUp(y);
+        assertTrue(result == x / y || result == x / y + 1);
+    }
+
+    function test_invariant_divRoundingUp(uint256 x, uint256 y) public {
         vm.assume(y != 0);
         uint256 z = x.divRoundingUp(y);
         uint256 diff = z - (x / y);