• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 use rayon::prelude::*;
2 
check<I>(iter: I) where I: ParallelIterator + Clone, I::Item: std::fmt::Debug + PartialEq,3 fn check<I>(iter: I)
4 where
5     I: ParallelIterator + Clone,
6     I::Item: std::fmt::Debug + PartialEq,
7 {
8     let a: Vec<_> = iter.clone().collect();
9     let b: Vec<_> = iter.collect();
10     assert_eq!(a, b);
11 }
12 
check_count<I>(iter: I) where I: ParallelIterator + Clone,13 fn check_count<I>(iter: I)
14 where
15     I: ParallelIterator + Clone,
16 {
17     assert_eq!(iter.clone().count(), iter.count());
18 }
19 
20 #[test]
clone_binary_heap()21 fn clone_binary_heap() {
22     use std::collections::BinaryHeap;
23     let heap: BinaryHeap<_> = (0..1000).collect();
24     check(heap.par_iter());
25     check(heap.into_par_iter());
26 }
27 
28 #[test]
clone_btree_map()29 fn clone_btree_map() {
30     use std::collections::BTreeMap;
31     let map: BTreeMap<_, _> = (0..1000).enumerate().collect();
32     check(map.par_iter());
33 }
34 
35 #[test]
clone_btree_set()36 fn clone_btree_set() {
37     use std::collections::BTreeSet;
38     let set: BTreeSet<_> = (0..1000).collect();
39     check(set.par_iter());
40 }
41 
42 #[test]
clone_hash_map()43 fn clone_hash_map() {
44     use std::collections::HashMap;
45     let map: HashMap<_, _> = (0..1000).enumerate().collect();
46     check(map.par_iter());
47 }
48 
49 #[test]
clone_hash_set()50 fn clone_hash_set() {
51     use std::collections::HashSet;
52     let set: HashSet<_> = (0..1000).collect();
53     check(set.par_iter());
54 }
55 
56 #[test]
clone_linked_list()57 fn clone_linked_list() {
58     use std::collections::LinkedList;
59     let list: LinkedList<_> = (0..1000).collect();
60     check(list.par_iter());
61     check(list.into_par_iter());
62 }
63 
64 #[test]
clone_vec_deque()65 fn clone_vec_deque() {
66     use std::collections::VecDeque;
67     let deque: VecDeque<_> = (0..1000).collect();
68     check(deque.par_iter());
69     check(deque.into_par_iter());
70 }
71 
72 #[test]
clone_option()73 fn clone_option() {
74     let option = Some(0);
75     check(option.par_iter());
76     check(option.into_par_iter());
77 }
78 
79 #[test]
clone_result()80 fn clone_result() {
81     let result = Ok::<_, ()>(0);
82     check(result.par_iter());
83     check(result.into_par_iter());
84 }
85 
86 #[test]
clone_range()87 fn clone_range() {
88     check((0..1000).into_par_iter());
89 }
90 
91 #[test]
clone_range_inclusive()92 fn clone_range_inclusive() {
93     check((0..=1000).into_par_iter());
94 }
95 
96 #[test]
clone_str()97 fn clone_str() {
98     let s = include_str!("clones.rs");
99     check(s.par_chars());
100     check(s.par_lines());
101     check(s.par_split('\n'));
102     check(s.par_split_inclusive('\n'));
103     check(s.par_split_terminator('\n'));
104     check(s.par_split_whitespace());
105     check(s.par_split_ascii_whitespace());
106 }
107 
108 #[test]
clone_vec()109 fn clone_vec() {
110     let v: Vec<_> = (0..1000).collect();
111     check(v.par_iter());
112     check(v.par_chunk_by(i32::eq));
113     check(v.par_chunks(42));
114     check(v.par_chunks_exact(42));
115     check(v.par_rchunks(42));
116     check(v.par_rchunks_exact(42));
117     check(v.par_windows(42));
118     check(v.par_split(|x| x % 3 == 0));
119     check(v.par_split_inclusive(|x| x % 3 == 0));
120     check(v.into_par_iter());
121 }
122 
123 #[test]
clone_array()124 fn clone_array() {
125     let a = [0i32; 100];
126     check(a.into_par_iter());
127 }
128 
129 #[test]
clone_adaptors()130 fn clone_adaptors() {
131     let v: Vec<_> = (0..1000).map(Some).collect();
132     check(v.par_iter().by_exponential_blocks());
133     check(v.par_iter().by_uniform_blocks(100));
134     check(v.par_iter().chain(&v));
135     check(v.par_iter().cloned());
136     check(v.par_iter().copied());
137     check(v.par_iter().enumerate());
138     check(v.par_iter().filter(|_| true));
139     check(v.par_iter().filter_map(|x| *x));
140     check(v.par_iter().flat_map(|x| *x));
141     check(v.par_iter().flat_map_iter(|x| *x));
142     check(v.par_iter().flatten());
143     check(v.par_iter().flatten_iter());
144     check(v.par_iter().with_max_len(1).fold(|| 0, |x, _| x));
145     check(v.par_iter().with_max_len(1).fold_with(0, |x, _| x));
146     check(v.par_iter().with_max_len(1).fold_chunks(1, || 0, |x, _| x));
147     check(
148         v.par_iter()
149             .with_max_len(1)
150             .fold_chunks_with(1, 0, |x, _| x),
151     );
152     check(v.par_iter().with_max_len(1).try_fold(|| 0, |_, &x| x));
153     check(v.par_iter().with_max_len(1).try_fold_with(0, |_, &x| x));
154     check(v.par_iter().inspect(|_| ()));
155     check(v.par_iter().update(|_| ()));
156     check(v.par_iter().interleave(&v));
157     check(v.par_iter().interleave_shortest(&v));
158     check(v.par_iter().intersperse(&None));
159     check(v.par_iter().chunks(3));
160     check(v.par_iter().map(|x| x));
161     check(v.par_iter().map_with(0, |_, x| x));
162     check(v.par_iter().map_init(|| 0, |_, x| x));
163     check(v.par_iter().panic_fuse());
164     check(v.par_iter().positions(|_| true));
165     check(v.par_iter().rev());
166     check(v.par_iter().skip(42));
167     check(v.par_iter().skip_any_while(|_| false));
168     check(v.par_iter().take(42));
169     check(v.par_iter().take_any_while(|_| true));
170     check(v.par_iter().cloned().while_some());
171     check(v.par_iter().with_max_len(1));
172     check(v.par_iter().with_min_len(1));
173     check(v.par_iter().zip(&v));
174     check(v.par_iter().zip_eq(&v));
175     check(v.par_iter().step_by(2));
176 }
177 
178 #[test]
clone_counted_adaptors()179 fn clone_counted_adaptors() {
180     let v: Vec<_> = (0..1000).collect();
181     check_count(v.par_iter().skip_any(42));
182     check_count(v.par_iter().take_any(42));
183 }
184 
185 #[test]
clone_empty()186 fn clone_empty() {
187     check(rayon::iter::empty::<i32>());
188 }
189 
190 #[test]
clone_once()191 fn clone_once() {
192     check(rayon::iter::once(10));
193 }
194 
195 #[test]
clone_repeat()196 fn clone_repeat() {
197     let x: Option<i32> = None;
198     check(rayon::iter::repeat(x).while_some());
199     check(rayon::iter::repeatn(x, 1000));
200 }
201 
202 #[test]
clone_splitter()203 fn clone_splitter() {
204     check(rayon::iter::split(0..1000, |x| (x, None)));
205 }
206 
207 #[test]
clone_multizip()208 fn clone_multizip() {
209     let v: &Vec<_> = &(0..1000).collect();
210     check((v,).into_par_iter());
211     check((v, v).into_par_iter());
212     check((v, v, v).into_par_iter());
213     check((v, v, v, v).into_par_iter());
214     check((v, v, v, v, v).into_par_iter());
215     check((v, v, v, v, v, v).into_par_iter());
216     check((v, v, v, v, v, v, v).into_par_iter());
217     check((v, v, v, v, v, v, v, v).into_par_iter());
218     check((v, v, v, v, v, v, v, v, v).into_par_iter());
219     check((v, v, v, v, v, v, v, v, v, v).into_par_iter());
220     check((v, v, v, v, v, v, v, v, v, v, v).into_par_iter());
221     check((v, v, v, v, v, v, v, v, v, v, v, v).into_par_iter());
222 }
223