feat: add more typed test assertions

PaulRBerg · Nov 27, 2022 · ddb0846 · ddb0846
1 parent c52a0de
commit ddb0846
Showing 1 changed file with 40 additions and 0 deletions.
diff --git 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);
+    }
 }