def test_on_delegate_class(self):
class AsyncIterator:
async def __anext__(self):
...
class AsyncIterable:
def __aiter__(self):
return AsyncIterator()
assert aio.is_async_iterable(AsyncIterable)
async def _wrap_iter(self) -> typing.AsyncGenerator[typing.Any, bytes]:
if isinstance(self.data, bytes):
for i in range(0, len(self.data), _MAGIC):
yield self.data[i:i + _MAGIC] # noqa: E203
elif aio.is_async_iterator(self.data) or inspect.isasyncgen(self.data):
try:
while True:
yield self._assert_bytes(
await
self.data.__anext__()) # type: ignore[union-attr]
except StopAsyncIteration:
pass
elif isinstance(self.data, typing.Iterator):
try:
while True:
yield self._assert_bytes(next(self.data))
except StopIteration:
pass
elif inspect.isgenerator(self.data):
try:
while True:
yield self._assert_bytes(
self.data.send(None)) # type: ignore[union-attr]
except StopIteration:
pass
elif aio.is_async_iterable(self.data):
async for chunk in self.data: # type: ignore[union-attr]
yield self._assert_bytes(chunk)
elif isinstance(self.data, typing.Iterable):
for chunk in self.data:
yield self._assert_bytes(chunk)
else:
# Will always fail.
self._assert_bytes(self.data)
def test_on_AsyncIterator_class(self):
class AsyncIterator:
async def __anext__(self):
return ...
assert not aio.is_async_iterable(AsyncIterator)
def test_on_class(self):
class AsyncIter:
async def __aiter__(self):
yield ...
assert aio.is_async_iterable(AsyncIter)
