Function test_pyproxy_iter

src/tests/test_pyproxy.py:211–265

(selenium)

@run_in_pyodide
def test_pyproxy_iter(selenium):
    from collections import ChainMap

    from pyodide.code import run_js

    def test_generator():
        yield from range(10)

    c = test_generator()
    ty, l = run_js("(c) => [c.type, [...c]]")(c)
    assert ty == "generator"
    assert l.to_py() == list(range(10))

    c = ChainMap({"a": 2, "b": 3})
    ty, l = run_js("(c) => [c.type, [...c]]")(c)
    assert ty == "ChainMap"
    assert set(l) == {"a", "b"}

    def test_generator2():
        acc = 0
        for i in range(10):
            r = yield acc
            acc += i * r

    c = test_generator2()

    result, result2 = run_js(
        """
        (c) => {
            let {done, value} = c.next();
            let result = [];
            while(!done){
                result.push(value);
                ({done, value} = c.next(value + 1));
            }

            function* test(){
                let acc = 0;
                for(let i=0; i < 10; i++){
                    let r = yield acc;
                    acc += i * r;
                }
            }
            c = test();
            ({done, value} = c.next());
            let result2 = [];
            while(!done){
                result2.push(value);
                ({done, value} = c.next(value + 1));
            }
            return [result, result2];
        }
        """
    )(c)
    assert result.to_py() == result2.to_py()


@run_in_pyodide