From 1eba240e7fbe5b01e7b82399939eec34684a884a Mon Sep 17 00:00:00 2001 From: Rappie Date: Fri, 1 Mar 2024 20:39:42 +0100 Subject: [PATCH] Add more types to assertion wrapper functions --- src/AssertWrapper.sol | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) diff --git a/src/AssertWrapper.sol b/src/AssertWrapper.sol index ed247bd..45f00a2 100644 --- a/src/AssertWrapper.sol +++ b/src/AssertWrapper.sol @@ -8,26 +8,54 @@ abstract contract AssertWrapper is AssertHelper { assertGt(a, b, message); } + function gt(int256 a, int256 b, string memory message) internal { + assertGt(a, b, message); + } + function lt(uint256 a, uint256 b, string memory message) internal { assertLt(a, b, message); } + function lt(int256 a, int256 b, string memory message) internal { + assertLt(a, b, message); + } + function gte(uint256 a, uint256 b, string memory message) internal { assertGte(a, b, message); } + function gte(int256 a, int256 b, string memory message) internal { + assertGte(a, b, message); + } + function lte(uint256 a, uint256 b, string memory message) internal { assertLte(a, b, message); } + function lte(int256 a, int256 b, string memory message) internal { + assertLte(a, b, message); + } + function eq(uint256 a, uint256 b, string memory message) internal { assertEq(a, b, message); } + function eq(int256 a, int256 b, string memory message) internal { + assertEq(a, b, message); + } + + function eq(bytes4 a, bytes4 b, string memory message) internal { + assertEq(a, b, message); + } + function neq(uint256 a, uint256 b, string memory message) internal { assertNeq(a, b, message); } + function neq(int256 a, int256 b, string memory message) internal { + assertNeq(a, b, message); + } + function t(bool a, string memory message) internal { assertWithMsg(a, message); }