yugen 0.1.9

Writing rust like python
Documentation 
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
94
95
96
97
98
99

pub mod pyrange {
    pub struct Pyrange {
        start: i64,
        end: i64,
        step: i64,
    }

    impl Pyrange {
        pub fn new(start: i64, end: i64, step: i64) -> Self {
            Pyrange { start, end, step }
        }
    }


    impl Iterator for Pyrange {
        type Item = i64;

        fn next(&mut self) -> Option<Self::Item> {
            if self.step == 0 {
            return None;
            }

            let next_value = self.start;
            if (self.step > 0 && next_value >= self.end) || (self.step < 0 && next_value <= self.end) {
                return None;
            }

            self.start += self.step;
            Some(next_value)
        }
    }
}

#[macro_export]
macro_rules! py_range {
    ($start:expr) => {{
         let end = $start;
        let start = 0;
         let step = 1;
         $crate::pyrange::Pyrange::new(start, end, step)
    }};
    ($start:expr, $end:expr) => {{
     let (start, end) = if $start <= $end {
             ($start, $end)
        } else {
            ($end, $start)
         };
        let step = 1;
        $crate::pyrange::Pyrange::new(start, end, step)
    }};
    ($start:expr, $end:expr, $step:expr) => {{
        let (start, end, step) = if $step > 0 {
            ($start, $end, $step)
        } else if $start <= $end {
         ($end, $start, $step)
        } else {
            ($start, $end, $step)
         };
         $crate::pyrange::Pyrange::new(start, end, step)
    }};
}




#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_py_range() {
        // 创建一个从0到9的迭代器，步长为1
        let mut range = py_range!(0, 10, 1);

        // 使用for循环遍历迭代器
        let mut values = Vec::new();
        for i in &mut range {
            values.push(i);
        }
        assert_eq!(values, vec![0, 1, 2, 3, 4, 5, 6, 7, 8, 9]);




        // 创建一个从0-9的迭代器，步长为-1
        let mut range_rev = py_range!(10, 0, -1);

        // 使用for循环遍历迭代器
        let mut values_rev = Vec::new();
        for i in &mut range_rev {
            values_rev.push(i);
        }
        assert_eq!(values_rev, vec![10,9, 8, 7, 6, 5, 4, 3, 2, 1]);


    }
}