Implement size_of for Felt (#123)

Author: dafnamatsry

crates/starknet-types-core/Cargo.toml

@@ -15,6 +15,7 @@ lambdaworks-math = { version = "0.10.0", default-features = false }
 num-traits = { version = "0.2", default-features = false }
 num-bigint = { version = "0.4", default-features = false }
 num-integer = { version = "0.1", default-features = false }
+size-of = { version = "0.1.5", default-features = false }
 
 # Optional
 arbitrary = { version = "1.3", optional = true }

crates/starknet-types-core/src/felt/mod.rs

@@ -8,6 +8,7 @@ use core::str::FromStr;
 use num_bigint::{BigInt, BigUint, Sign};
 use num_integer::Integer;
 use num_traits::{One, Zero};
+use size_of::SizeOf;
 #[cfg(any(feature = "prime-bigint", test))]
 use {lazy_static::lazy_static, num_traits::Num};
 
@@ -44,6 +45,10 @@ use arbitrary::{self, Arbitrary, Unstructured};
 #[derive(Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
 pub struct Felt(pub(crate) FieldElement<Stark252PrimeField>);
 
+impl SizeOf for Felt {
+    fn size_of_children(&self, _context: &mut size_of::Context) {}
+}
+
 /// A non-zero [Felt].
 #[repr(transparent)]
 #[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
@@ -1072,6 +1077,7 @@ mod test {
     use core::ops::Shl;
     use proptest::prelude::*;
     use regex::Regex;
+    use size_of::TotalSize;
 
     #[test]
     fn test_debug_format() {
@@ -1856,4 +1862,15 @@ mod test {
         felt.zeroize();
         assert_eq!(felt, Felt::ZERO);
     }
+
+    #[test]
+    fn felt_size_of() {
+        assert_eq!(Felt::ZERO.size_of(), TotalSize::total(32));
+        assert_eq!(Felt::ONE.size_of(), TotalSize::total(32));
+        assert_eq!(
+            Felt(FieldElement::from(1600000000)).size_of(),
+            TotalSize::total(32)
+        );
+        assert_eq!(Felt::MAX.size_of(), TotalSize::total(32));
+    }
 }