1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
mod gen_range;
mod combi;
pub struct RangeIterator {
curr: i32,
stop: i32,
step: i32,
}
impl RangeIterator {
pub fn new(start: i32, stop: i32, step: i32) -> Self {
RangeIterator {
curr: start,
stop,
step,
}
}
}
impl Iterator for RangeIterator {
type Item = i32;
fn next(&mut self) -> Option<Self::Item> {
if self.curr >= self.stop {
return None;
}
let res = self.curr;
self.curr += self.step;
Some(res)
}
}
#[cfg(test)]
mod tests {
use super::RangeIterator;
use itertools::Itertools;
#[test]
fn it_works() {
assert_eq!(2 + 2, 4);
}
#[test]
fn iter_test() {
let my_iter = RangeIterator::new(2, 12, 2);
let mut sum = 0;
for s in my_iter {
println!("s, sum = {}, {}", s, sum);
sum += s;
}
assert_eq!(sum, 30, "Test 1");
}
#[test]
fn tup_test() {
let mut v = Vec::new();
for (a, b) in (1..5).tuples() {
v.push((a, b));
}
assert_eq!(v, vec![(1, 2), (3, 4)]);
let mut it = (1..7).tuples();
assert_eq!(Some((1, 2, 3)), it.next());
assert_eq!(Some((4, 5, 6)), it.next());
assert_eq!(None, it.next());
let it = (1..7).tuples::<(_, _, _)>();
itertools::assert_equal(it, vec![(1, 2, 3), (4, 5, 6)]);
use itertools::Tuples;
use std::ops::Range;
let it: Tuples<Range<u32>, (u32, u32, u32)> = (1..7).tuples();
itertools::assert_equal(it, vec![(1, 2, 3), (4, 5, 6)]);
}
}