diff --git a/src/test/Assertions.sol b/src/test/Assertions.sol index ca5b73e9..df79b7d3 100644 --- a/src/test/Assertions.sol +++ b/src/test/Assertions.sol @@ -11,7 +11,47 @@ contract Assertions is PRBTest { assertEq(SD59x18.unwrap(a), SD59x18.unwrap(b)); } + function assertEq(SD59x18 a, SD59x18 b, string memory err) internal { + assertEq(SD59x18.unwrap(a), SD59x18.unwrap(b), err); + } + + function assertEq(SD59x18 a, int256 b) internal { + assertEq(SD59x18.unwrap(a), b); + } + + function assertEq(SD59x18 a, int256 b, string memory err) internal { + assertEq(SD59x18.unwrap(a), b, err); + } + + function assertEq(int256 a, SD59x18 b) internal { + assertEq(a, SD59x18.unwrap(b)); + } + + function assertEq(int256 a, SD59x18 b, string memory err) internal { + assertEq(a, SD59x18.unwrap(b), err); + } + function assertEq(UD60x18 a, UD60x18 b) internal { assertEq(UD60x18.unwrap(a), UD60x18.unwrap(b)); } + + function assertEq(UD60x18 a, UD60x18 b, string memory err) internal { + assertEq(UD60x18.unwrap(a), UD60x18.unwrap(b), err); + } + + function assertEq(UD60x18 a, uint256 b) internal { + assertEq(UD60x18.unwrap(a), b); + } + + function assertEq(UD60x18 a, uint256 b, string memory err) internal { + assertEq(UD60x18.unwrap(a), b, err); + } + + function assertEq(uint256 a, UD60x18 b) internal { + assertEq(a, UD60x18.unwrap(b)); + } + + function assertEq(uint256 a, UD60x18 b, string memory err) internal { + assertEq(a, UD60x18.unwrap(b), err); + } }