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
|
import os
import pytest
import parsl
from parsl.app.app import python_app
from parsl.tests.configs.local_threads import config
def local_setup():
global dfk
dfk = parsl.load(config)
def local_teardown():
parsl.dfk().cleanup()
@python_app
def slow_double(x, sleep_dur=1, cache=True):
import time
time.sleep(sleep_dur)
return x * 2
@pytest.mark.local
def test_checkpointing():
"""Testing code snippet from documentation
"""
N = 5 # Number of calls to slow_double
d = [] # List to store the futures
for i in range(0, N):
d.append(slow_double(i))
# Wait for the results
[i.result() for i in d]
checkpoint_dir = dfk.checkpoint()
print(checkpoint_dir)
assert os.path.exists(checkpoint_dir), "Checkpoint dir does not exist"
|
