Simplify some test cases

js2xxx · Feb 5, 2025 · dcb16a4 · dcb16a4
1 parent 12f695c
commit dcb16a4
Show file tree

Hide file tree

Showing 4 changed files with 42 additions and 29 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/examples/exchange.rs b/examples/exchange.rs
@@ -116,7 +116,7 @@ impl<'a, A: Allocator> Session<'a, A> {
             let iter = (0..BATCH_SIZE).map(|index| {
                 let size = 64;
                 let mut vec = Vec::with_capacity_in(size, self.alloc);
-                vec.extend(iter::repeat(index as u8).take(size.min(128)));
+                vec.extend(iter::repeat_n(index as u8, size.min(128)));
                 vec.into_boxed_slice()
             });
             let mut vec = Vec::with_capacity_in(BATCH_SIZE, self.alloc);

diff --git a/examples/random2.rs b/examples/random2.rs
@@ -103,7 +103,7 @@ fn warm_up<'a, A: Allocator>(array: &mut [Option<Vec<u8, &'a A>>], a: &'a A) {
 
 fn new_vec<A: Allocator>(size: usize, a: A) -> Vec<u8, A> {
     let mut v = Vec::with_capacity_in(size, a);
-    v.extend(iter::repeat(0).take(size));
+    v.extend(iter::repeat_n(0, size));
     v
 }
 

diff --git a/src/lib.rs b/src/lib.rs
@@ -224,7 +224,7 @@ mod test {
 
     const CHUNK_SIZE: usize = 1024;
     fn chunks(size: usize) -> impl Iterator<Item = [u8; CHUNK_SIZE]> {
-        iter::repeat([0u8; CHUNK_SIZE]).take(size / CHUNK_SIZE)
+        iter::repeat_n([0u8; CHUNK_SIZE], size / CHUNK_SIZE)
     }
 
     #[test]
@@ -257,8 +257,8 @@ mod test {
     fn multithread() {
         let j = thread::spawn(move || {
             let mut vec = Vec::new_in(Ferroc);
-            vec.extend(iter::repeat(0u8).take(100));
-            vec.extend(iter::repeat(1u8).take(100));
+            vec.extend(iter::repeat_n(0u8, 100));
+            vec.extend(iter::repeat_n(1u8, 100));
             vec
         });
         let vec = j.join().unwrap();