chore: implement felt! for all uint types

Signed-off-by: Dori Medini <[email protected]>
starkware-libs · Jun 3, 2024 · 008dbf5 · 008dbf5
1 parent 82fd726
commit 008dbf5
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/src/hash.rs b/src/hash.rs
@@ -26,13 +26,23 @@ pub trait TryIntoFelt<V> {
     fn to_felt_unchecked(v: V) -> Felt;
 }
 
-#[cfg(any(feature = "testing", test))]
-impl TryIntoFelt<u128> for FeltConverter {
-    fn to_felt_unchecked(v: u128) -> Felt {
-        Felt::from(v)
-    }
+macro_rules! impl_try_into_felt {
+    ($type:ty) => {
+        #[cfg(any(feature = "testing", test))]
+        impl TryIntoFelt<$type> for FeltConverter {
+            fn to_felt_unchecked(v: $type) -> Felt {
+                Felt::from(v)
+            }
+        }
+    };
 }
 
+impl_try_into_felt!(u128);
+impl_try_into_felt!(u64);
+impl_try_into_felt!(u32);
+impl_try_into_felt!(u16);
+impl_try_into_felt!(u8);
+
 #[cfg(any(feature = "testing", test))]
 impl TryIntoFelt<&str> for FeltConverter {
     fn to_felt_unchecked(v: &str) -> Felt {