Implement Felt from bool

lambdaclass · Dec 12, 2023 · 4bc8a5d · 4bc8a5d
1 parent ebfebde
commit 4bc8a5d
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/crates/starknet-types-core/src/felt.rs b/crates/starknet-types-core/src/felt.rs
@@ -444,6 +444,15 @@ impl From<i128> for Felt {
     }
 }
 
+impl From<bool> for Felt {
+    fn from(value: bool) -> Felt {
+        match value {
+            true => Felt::ONE,
+            false => Felt::ZERO,
+        }
+    }
+}
+
 macro_rules! impl_from {
     ($from:ty, $with:ty) => {
         impl From<$from> for Felt {
@@ -1698,4 +1707,11 @@ mod test {
             );
         }
     }
+    #[test]
+    fn bool_into_felt() {
+        let zero: Felt = false.into();
+        let one: Felt = true.into();
+        assert_eq!(one, Felt::ONE);
+        assert_eq!(zero, Felt::ZERO);
+    }
 }