1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
import pytest
from aiostream.test_utils import event_loop, add_resource
from aiostream import stream, streamcontext, operator
# Pytest fixtures
event_loop
@pytest.mark.asyncio
async def test_streamcontext(event_loop):
with event_loop.assert_cleanup():
xs = stream.range(3) | add_resource.pipe(1)
async with streamcontext(xs) as streamer:
it = iter(range(3))
async for item in streamer:
assert item == next(it)
assert event_loop.steps == [1]
with event_loop.assert_cleanup():
xs = stream.range(5) | add_resource.pipe(1)
async with xs.stream() as streamer:
it = iter(range(5))
async for item in streamer:
assert item == next(it)
assert event_loop.steps == [1]
def test_operator_from_method():
with pytest.raises(ValueError):
class A:
@operator
async def method(self, arg):
yield 1
with pytest.raises(ValueError):
class B:
@operator
async def method(cls, arg):
yield 1
with pytest.raises(ValueError):
class C:
@operator
@classmethod
async def method(cls, arg):
yield 1
@pytest.mark.asyncio
async def test_error_on_sync_iteration(event_loop):
xs = stream.range(3)
# Stream raises a TypeError
with pytest.raises(TypeError):
for x in xs:
assert False
# Streamer raises a TypeError
async with xs.stream() as streamer:
with pytest.raises(TypeError):
for x in streamer:
assert False
@pytest.mark.asyncio
async def test_error_on_entering_a_stream(event_loop):
xs = stream.range(3)
# Stream raises a TypeError
with pytest.raises(TypeError) as ctx:
async with xs:
assert False
assert "Use the `stream` method" in str(ctx.value)
def test_compatibility():
@operator
async def test1():
yield 1
with pytest.raises(AttributeError):
test1.pipe
match = "The `pipable` argument is deprecated."
with pytest.warns(DeprecationWarning, match=match):
@operator()
async def test2():
yield 1
with pytest.raises(AttributeError):
test2.pipe
with pytest.warns(DeprecationWarning, match=match):
@operator(pipable=False)
async def test3():
yield 1
with pytest.raises(AttributeError):
test3.pipe
with pytest.warns(DeprecationWarning, match=match):
@operator(pipable=True)
async def test4(source):
yield 1
test4.pipe
|
