|
1 |
| -from datetime import datetime |
2 |
| -import pytest |
3 | 1 | import asyncio
|
4 | 2 |
|
| 3 | +import pytest |
5 | 4 |
|
6 | 5 | from python_utils import types
|
7 |
| -from python_utils.aio import acount, acontainer |
| 6 | +from python_utils.aio import acontainer, acount, adict |
8 | 7 |
|
9 | 8 |
|
10 | 9 | @pytest.mark.asyncio
|
11 |
| -async def test_acount(monkeypatch: pytest.MonkeyPatch): |
| 10 | +async def test_acount(monkeypatch: pytest.MonkeyPatch) -> None: |
12 | 11 | sleeps: types.List[float] = []
|
13 | 12 |
|
14 |
| - async def mock_sleep(delay: float): |
| 13 | + async def mock_sleep(delay: float) -> None: |
15 | 14 | sleeps.append(delay)
|
16 | 15 |
|
17 | 16 | monkeypatch.setattr(asyncio, 'sleep', mock_sleep)
|
18 | 17 |
|
19 |
| - async for i in acount(delay=1, stop=3.5): |
20 |
| - print('i', i, datetime.now()) |
| 18 | + async for _i in acount(delay=1, stop=3.5): |
| 19 | + pass |
21 | 20 |
|
22 | 21 | assert len(sleeps) == 4
|
23 | 22 | assert sum(sleeps) == 4
|
24 | 23 |
|
25 | 24 |
|
26 | 25 | @pytest.mark.asyncio
|
27 |
| -async def test_acontainer(): |
28 |
| - async def async_gen(): |
| 26 | +async def test_acontainer() -> None: |
| 27 | + async def async_gen() -> types.AsyncIterable[int]: |
29 | 28 | yield 1
|
30 | 29 | yield 2
|
31 | 30 | yield 3
|
32 | 31 |
|
33 |
| - async def empty_gen(): |
| 32 | + async def empty_gen() -> types.AsyncIterable[int]: |
34 | 33 | if False:
|
35 | 34 | yield 1
|
36 | 35 |
|
37 | 36 | assert await acontainer(async_gen) == [1, 2, 3]
|
38 | 37 | assert await acontainer(async_gen()) == [1, 2, 3]
|
39 | 38 | assert await acontainer(async_gen, set) == {1, 2, 3}
|
40 | 39 | assert await acontainer(async_gen(), set) == {1, 2, 3}
|
| 40 | + assert await acontainer(async_gen, list) == [1, 2, 3] |
| 41 | + assert await acontainer(async_gen(), list) == [1, 2, 3] |
| 42 | + assert await acontainer(async_gen, tuple) == (1, 2, 3) |
| 43 | + assert await acontainer(async_gen(), tuple) == (1, 2, 3) |
41 | 44 | assert await acontainer(empty_gen) == []
|
42 | 45 | assert await acontainer(empty_gen()) == []
|
| 46 | + assert await acontainer(empty_gen, set) == set() |
| 47 | + assert await acontainer(empty_gen(), set) == set() |
| 48 | + assert await acontainer(empty_gen, list) == list() |
| 49 | + assert await acontainer(empty_gen(), list) == list() |
| 50 | + assert await acontainer(empty_gen, tuple) == tuple() |
| 51 | + assert await acontainer(empty_gen(), tuple) == tuple() |
| 52 | + |
| 53 | + |
| 54 | +@pytest.mark.asyncio |
| 55 | +async def test_adict() -> None: |
| 56 | + async def async_gen() -> types.AsyncIterable[types.Tuple[int, int]]: |
| 57 | + yield 1, 2 |
| 58 | + yield 3, 4 |
| 59 | + yield 5, 6 |
| 60 | + |
| 61 | + async def empty_gen() -> types.AsyncIterable[types.Tuple[int, int]]: |
| 62 | + if False: |
| 63 | + yield 1, 2 |
| 64 | + |
| 65 | + assert await adict(async_gen) == {1: 2, 3: 4, 5: 6} |
| 66 | + assert await adict(async_gen()) == {1: 2, 3: 4, 5: 6} |
| 67 | + assert await adict(empty_gen) == {} |
| 68 | + assert await adict(empty_gen()) == {} |
0 commit comments