wehub-resource-sync
71 lines
1.4 KiB
Python
71 lines
1.4 KiB
Python
import sys
|
|
from collections.abc import AsyncIterator
|
|
|
|
import pytest
|
|
|
|
from ray._private.async_compat import has_async_methods, is_async_func, sync_to_async
|
|
|
|
|
|
def test_is_async_func():
|
|
def f():
|
|
return 1
|
|
|
|
def f_gen():
|
|
yield 1
|
|
|
|
async def g():
|
|
return 1
|
|
|
|
async def g_gen():
|
|
yield 1
|
|
|
|
assert is_async_func(f) is False
|
|
assert is_async_func(f_gen) is False
|
|
assert is_async_func(g) is True
|
|
assert is_async_func(g_gen) is True
|
|
|
|
|
|
class NoAsyncMethods:
|
|
def sync_fn(self) -> None:
|
|
pass
|
|
|
|
|
|
class YesAsyncMethods:
|
|
async def async_fn(self) -> None:
|
|
pass
|
|
|
|
|
|
class YesAsyncGenMethods:
|
|
async def async_gen_fn(self) -> AsyncIterator[None]:
|
|
yield None
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"cls, expected",
|
|
[
|
|
(NoAsyncMethods, False),
|
|
(YesAsyncMethods, True),
|
|
(YesAsyncGenMethods, True),
|
|
],
|
|
)
|
|
def test_has_async_methods(cls, expected: bool) -> None:
|
|
assert has_async_methods(cls) is expected
|
|
|
|
|
|
def test_sync_to_async_is_cached() -> None:
|
|
def sync_fn() -> None:
|
|
pass
|
|
|
|
async def async_fn() -> None:
|
|
pass
|
|
|
|
assert sync_to_async(sync_fn) is sync_to_async(sync_fn)
|
|
assert sync_to_async(async_fn) is sync_to_async(async_fn)
|
|
|
|
assert sync_to_async(sync_fn) is not sync_to_async(async_fn)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
sys.exit(pytest.main(["-sv", __file__]))
|