rust/tests/ui/iterators/generator_capture.rs
Oli Scherer 5fbdfc3e10
Add iter macro
This adds an `iter!` macro that can be used to create movable
generators.

This also adds a yield_expr feature so the `yield` keyword can be used
within iter! macro bodies. This was needed because several unstable
features each need `yield` expressions, so this allows us to stabilize
them separately from any individual feature.

Co-authored-by: Oli Scherer <github35764891676564198441@oli-obk.de>
Co-authored-by: Jieyou Xu <jieyouxu@outlook.com>
Co-authored-by: Travis Cross <tc@traviscross.com>
2025-06-03 10:52:32 -07:00

28 lines
607 B
Rust

//@ run-pass
#![feature(iter_macro, yield_expr)]
use std::iter::iter;
fn main() {
let i = {
let s = String::new();
iter! { move || {
yield s.len();
for x in 5..10 {
yield x * 2;
}
}}
};
let mut i = i();
assert_eq!(i.next(), Some(0));
assert_eq!(i.next(), Some(10));
assert_eq!(i.next(), Some(12));
assert_eq!(i.next(), Some(14));
assert_eq!(i.next(), Some(16));
assert_eq!(i.next(), Some(18));
assert_eq!(i.next(), None);
assert_eq!(i.next(), None);
assert_eq!(i.next(), None);
}