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
|
"""Functions used to explain kwargs"""
from pathlib import Path
import pytest
from parsl import File, python_app
def test_inputs():
@python_app()
def map_app(x):
return x * 2
@python_app()
def reduce_app(inputs=()):
return sum(inputs)
map_futures = [map_app(x) for x in range(3)]
reduce_future = reduce_app(inputs=map_futures)
assert reduce_future.result() == 6
@pytest.mark.shared_fs
def test_outputs(tmpd_cwd):
@python_app()
def write_app(message, outputs=()):
"""Write a single message to every file in outputs"""
for path in outputs:
with open(path, 'w') as fp:
print(message, file=fp)
to_write = [
File(Path(tmpd_cwd) / 'output-0.txt'),
File(Path(tmpd_cwd) / 'output-1.txt')
]
write_app('Hello!', outputs=to_write).result()
for path in to_write:
with open(path) as fp:
assert fp.read() == 'Hello!\n'
|
