test: update test_iter_adapter_map

starkware-libs · Jan 4, 2025 · bde6cf1 · bde6cf1
1 parent 874e64f
commit bde6cf1
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/corelib/src/test/iter_test.cairo b/corelib/src/test/iter_test.cairo
@@ -2,9 +2,11 @@ use crate::iter::{IntoIterator, Iterator};
 
 #[test]
 fn test_iter_adapter_map() {
-    let mut i = 1;
-    for elem in (1..4_u8).into_iter().map(|x| 2 * x).map(|x| x + 1) {
-        assert_eq!(elem, i * 2 + 1);
-        i += 1;
-    }
+    let a = array![1, 2, 3];
+    let mut iter = a.into_iter().map(|x| 2 * x);
+
+    assert_eq!(iter.next(), Option::Some(2));
+    assert_eq!(iter.next(), Option::Some(4));
+    assert_eq!(iter.next(), Option::Some(6));
+    assert_eq!(iter.next(), Option::None);
 }