[KGA-3] [KGA-122] fix: SSJ pow (#1022)

kkrt-labs · Nov 13, 2024 · a24ea25 · a24ea25
1 parent 935c223
commit a24ea25
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/crates/utils/src/math.cairo b/crates/utils/src/math.cairo
@@ -38,6 +38,9 @@ impl ExponentiationImpl<
 > of Exponentiation<T> {
     fn pow(self: T, mut exponent: T) -> T {
         let zero = Zero::zero();
+        if exponent.is_zero() {
+            return One::one();
+        }
         if self.is_zero() {
             return zero;
         }
@@ -361,6 +364,7 @@ mod tests {
 
     #[test]
     fn test_wrapping_pow() {
+        assert(0_u256.wrapping_pow(0) == 1, '0^0 should be 1');
         assert(5_u256.wrapping_pow(10) == 9765625, '5^10 should be 9765625');
         assert(
             5_u256
@@ -376,6 +380,7 @@ mod tests {
 
     #[test]
     fn test_pow() {
+        assert(0_u256.pow(0) == 1, 'n^0 should be 1');
         assert(5_u256.pow(10) == 9765625, '5^10 should be 9765625');
         assert(5_u256.pow(45) == 28421709430404007434844970703125, '5^45 failed');
         assert(123456_u256.pow(0) == 1, 'n^0 should be 1');