Python generators vs Rust streams

Cargo.toml

[package]
name = "learn-async-stream"
version = "0.1.0"
edition = "2018"

[dependencies]
futures = "0.3"
futures-async-stream = "0.2"
tokio = { version = "0.2", features = ["full"] }

main.py

def foo():
    for i in range(10):
        yield i

for i in foo():
    print(i)

main.rs

#![feature(generators, proc_macro_hygiene, stmt_expr_attributes)]

use futures::stream::Stream;
use futures::pin_mut;
use futures_async_stream::stream;
use tokio::stream::StreamExt;

fn foo() -> impl Stream<Item = i32> {
    #[stream]
    async move {
        for i in 0..10 {
            yield i;
        }
    }
}

#[tokio::main]
async fn main() {
    let stream = foo();
    pin_mut!(stream);
    while let Some(s) = stream.next().await {
        println!("{}", s);
    }
}

https://github.com/taiki-e/futures-async-stream

https://docs.rs/futures-async-stream/0.2.0/futures_async_stream/