mirror of
https://github.com/pyscript/pyscript.git
synced 2025-12-19 18:27:29 -05:00
8c5475f78f
This PR adds support for optionally running pyodide in a web worker: - add a new option config.execution_thread, which can be `main` or `worker`. The default is `main` - improve the test machinery so that we run all tests twice, once for `main` and once for `worker` - add a new esbuild target which builds the code for the worker The support for workers is not complete and many features are still missing: there are 71 tests which are marked as `@skip_worker`, but we can fix them in subsequent PRs. The vast majority of tests fail because js.document is unavailable: for it to run transparently, we need the "auto-syncify" feature of synclink. Co-authored-by: Hood Chatham <roberthoodchatham@gmail.com> Co-authored-by: Madhur Tandon <20173739+madhur-tandon@users.noreply.github.com>
204 lines
6.4 KiB
Python
204 lines
6.4 KiB
Python
from .support import PyScriptTest, skip_worker
|
|
|
|
|
|
class TestAsync(PyScriptTest):
|
|
# ensure_future() and create_task() should behave similarly;
|
|
# we'll use the same source code to test both
|
|
coroutine_script = """
|
|
<py-script>
|
|
import js
|
|
import asyncio
|
|
js.console.log("first")
|
|
async def main():
|
|
await asyncio.sleep(1)
|
|
js.console.log("third")
|
|
asyncio.{func}(main())
|
|
js.console.log("second")
|
|
</py-script>
|
|
"""
|
|
|
|
def test_asyncio_ensure_future(self):
|
|
self.pyscript_run(self.coroutine_script.format(func="ensure_future"))
|
|
self.wait_for_console("third")
|
|
assert self.console.log.lines[-3:] == ["first", "second", "third"]
|
|
|
|
def test_asyncio_create_task(self):
|
|
self.pyscript_run(self.coroutine_script.format(func="create_task"))
|
|
self.wait_for_console("third")
|
|
assert self.console.log.lines[-3:] == ["first", "second", "third"]
|
|
|
|
def test_asyncio_gather(self):
|
|
self.pyscript_run(
|
|
"""
|
|
<py-script id="pys">
|
|
import asyncio
|
|
import js
|
|
from pyodide.ffi import to_js
|
|
|
|
async def coro(delay):
|
|
await asyncio.sleep(delay)
|
|
return(delay)
|
|
|
|
async def get_results():
|
|
results = await asyncio.gather(*[coro(d) for d in range(3,0,-1)])
|
|
js.console.log(str(results)) #Compare to string representation, not Proxy
|
|
js.console.log("DONE")
|
|
|
|
asyncio.ensure_future(get_results())
|
|
</py-script>
|
|
"""
|
|
)
|
|
self.wait_for_console("DONE")
|
|
assert self.console.log.lines[-2:] == ["[3, 2, 1]", "DONE"]
|
|
|
|
def test_multiple_async(self):
|
|
self.pyscript_run(
|
|
"""
|
|
<py-script>
|
|
import js
|
|
import asyncio
|
|
async def a_func():
|
|
for i in range(3):
|
|
js.console.log('A', i)
|
|
await asyncio.sleep(0.1)
|
|
asyncio.ensure_future(a_func())
|
|
</py-script>
|
|
|
|
<py-script>
|
|
import js
|
|
import asyncio
|
|
async def b_func():
|
|
for i in range(3):
|
|
js.console.log('B', i)
|
|
await asyncio.sleep(0.1)
|
|
js.console.log('b func done')
|
|
asyncio.ensure_future(b_func())
|
|
</py-script>
|
|
"""
|
|
)
|
|
self.wait_for_console("b func done")
|
|
assert self.console.log.lines == [
|
|
"A 0",
|
|
"B 0",
|
|
"A 1",
|
|
"B 1",
|
|
"A 2",
|
|
"B 2",
|
|
"b func done",
|
|
]
|
|
|
|
@skip_worker("FIXME: display()")
|
|
def test_multiple_async_multiple_display_targeted(self):
|
|
self.pyscript_run(
|
|
"""
|
|
<py-script id='pyA'>
|
|
import js
|
|
import asyncio
|
|
|
|
async def a_func():
|
|
for i in range(2):
|
|
display(f'A{i}', target='pyA')
|
|
await asyncio.sleep(0.1)
|
|
asyncio.ensure_future(a_func())
|
|
|
|
</py-script>
|
|
<py-script id='pyB'>
|
|
import js
|
|
import asyncio
|
|
|
|
async def a_func():
|
|
for i in range(2):
|
|
display(f'B{i}', target='pyB')
|
|
await asyncio.sleep(0.1)
|
|
js.console.log("B DONE")
|
|
|
|
asyncio.ensure_future(a_func())
|
|
</py-script>
|
|
"""
|
|
)
|
|
self.wait_for_console("B DONE")
|
|
inner_text = self.page.inner_text("html")
|
|
assert "A0\nA1\nB0\nB1" in inner_text
|
|
|
|
@skip_worker("FIXME: display()")
|
|
def test_async_display_untargeted(self):
|
|
self.pyscript_run(
|
|
"""
|
|
<py-script id='pyA'>
|
|
import asyncio
|
|
import js
|
|
|
|
async def a_func():
|
|
try:
|
|
display('A')
|
|
await asyncio.sleep(0.1)
|
|
except Exception as err:
|
|
js.console.error(str(err))
|
|
await asyncio.sleep(1)
|
|
js.console.log("DONE")
|
|
|
|
asyncio.ensure_future(a_func())
|
|
</py-script>
|
|
"""
|
|
)
|
|
self.wait_for_console("DONE")
|
|
assert (
|
|
self.console.error.lines[-1]
|
|
== "Implicit target not allowed here. Please use display(..., target=...)"
|
|
)
|
|
|
|
def test_sync_and_async_order(self):
|
|
"""
|
|
The order of execution is defined as follows:
|
|
1. first, we execute all the py-script tag in order
|
|
2. then, we start all the tasks which were scheduled with create_task
|
|
|
|
Note that tasks are started *AFTER* all py-script tags have been
|
|
executed. That's why the console.log() inside mytask1 and mytask2 are
|
|
executed after e.g. js.console.log("6").
|
|
"""
|
|
src = """
|
|
<py-script>
|
|
import js
|
|
js.console.log("1")
|
|
</py-script>
|
|
|
|
<py-script>
|
|
import asyncio
|
|
import js
|
|
|
|
async def mytask1():
|
|
js.console.log("7")
|
|
await asyncio.sleep(0)
|
|
js.console.log("9")
|
|
|
|
js.console.log("2")
|
|
asyncio.create_task(mytask1())
|
|
js.console.log("3")
|
|
</py-script>
|
|
|
|
<py-script>
|
|
import js
|
|
js.console.log("4")
|
|
</py-script>
|
|
|
|
<py-script>
|
|
import asyncio
|
|
import js
|
|
|
|
async def mytask2():
|
|
js.console.log("8")
|
|
await asyncio.sleep(0)
|
|
js.console.log("10")
|
|
js.console.log("DONE")
|
|
|
|
js.console.log("5")
|
|
asyncio.create_task(mytask2())
|
|
js.console.log("6")
|
|
</py-script>
|
|
"""
|
|
self.pyscript_run(src, wait_for_pyscript=False)
|
|
self.wait_for_console("DONE")
|
|
lines = self.console.log.lines[-11:]
|
|
assert lines == ["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "DONE"]
|