added calculation for closure type size

corelib/src/test/language_features/closure_test.cairo

@@ -57,15 +57,35 @@ fn option_map_test() {
     assert_eq!(option_map(Option::Some(2), |x| Option::Some(x)), Option::Some(Option::Some(2)));
 }
 
-fn array_map<T, F, impl Fn: core::ops::Fn<F, (T,)>, +Drop<T>, +Drop<F>, +Drop<Fn::Output>>(
+fn fix_sized_array_map<
+    T, F, impl Fn: core::ops::Fn<F, (T,)>, +Drop<T>, +Drop<F>, +Drop<Fn::Output>,
+>(
     arr: [T; 2], f: F,
 ) -> [core::ops::Fn::<F, (T,)>::Output; 2] {
     let [a, b] = arr;
     [f(a), f(b)]
 }
 
 #[test]
-fn array_map_test() {
-    assert_eq!(array_map([2, 3], |x| x + 3), [5, 6]);
+fn fix_sized_array_map_test() {
+    assert_eq!(fix_sized_array_map([2, 3], |x| x + 3), [5, 6]);
 }
 
+#[generate_trait]
+impl ArrayExt of ArrayExtTrait {
+    fn map<T, +Drop<T>, F, +Drop<F>, impl func: core::ops::Fn<F, (T,)>, +Drop<func::Output>>(
+        self: Array<T>, f: F,
+    ) -> Array<func::Output> {
+        let mut output: Array<func::Output> = array![];
+        for elem in self {
+            output.append(f(elem));
+        };
+        output
+    }
+}
+#[test]
+fn array_map_test() {
+    let arr = array![1, 2, 3];
+    let result = arr.map(|x| x + 1);
+    assert_eq!(result, array![2, 3, 4]);
+}

crates/cairo-lang-lowering/src/db.rs

@@ -798,7 +798,9 @@ fn type_size(db: &dyn LoweringGroup, ty: TypeId) -> usize {
                     .to_usize()
                     .unwrap()
         }
-        TypeLongId::Closure(_) => unimplemented!(),
+        TypeLongId::Closure(closure_ty) => {
+            closure_ty.captured_types.iter().map(|ty| db.type_size(*ty)).sum()
+        }
         TypeLongId::Coupon(_) => 0,
         TypeLongId::GenericParameter(_)
         | TypeLongId::Var(_)