/
test_dump_result.py
85 lines (62 loc) · 2.43 KB
/
test_dump_result.py
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
"""Test the dump_result convergence routine"""
import copy
import os
import pytest
import krotov
def incl_range(a, b, step=1):
e = 1 if step > 0 else -1
return range(a, b + e, step)
@pytest.fixture
def oct_result_20(request):
testdir = os.path.splitext(request.module.__file__)[0]
filename = os.path.join(testdir, "oct_result.dump")
return krotov.result.Result.load(filename)
@pytest.fixture
def oct_result_5(oct_result_20):
result = copy.deepcopy(oct_result_20)
result.iters = list(incl_range(0, 5))
return result
@pytest.fixture
def oct_result_6(oct_result_20):
result = copy.deepcopy(oct_result_20)
result.iters = list(incl_range(0, 6))
return result
@pytest.fixture
def oct_result_10(oct_result_20):
result = copy.deepcopy(oct_result_20)
result.iters = list(incl_range(0, 10))
return result
def test_invalid_dump_result(oct_result_20):
with pytest.raises(ValueError):
krotov.convergence.dump_result('oct_result.dump', every=0)
filename = "non_existent_folder/oct_result.dump"
dump_result = krotov.convergence.dump_result(filename)
response = dump_result(oct_result_20)
assert "Could not store" in response
def test_dump_result_overwrite(oct_result_5, oct_result_10, tmpdir):
filename = str(tmpdir.join("oct_result_a.dump"))
assert not os.path.isfile(filename)
dump_result = krotov.convergence.dump_result(filename, every=5)
response = dump_result(oct_result_5)
assert response is None
assert os.path.isfile(filename)
result = krotov.result.Result.load(filename)
assert result.iters[-1] == 5
response = dump_result(oct_result_10)
assert response is None
assert os.path.isfile(filename)
result = krotov.result.Result.load(filename)
assert result.iters[-1] == 10
def test_dump_result_keep(oct_result_5, oct_result_6, oct_result_10, tmpdir):
filename = str(tmpdir.join("oct_result_{iter:06d}.dump"))
assert not os.path.isfile(filename)
dump_result = krotov.convergence.dump_result(filename, every=5)
response = dump_result(oct_result_5)
assert response is None
assert os.path.isfile(str(tmpdir.join("oct_result_000005.dump")))
response = dump_result(oct_result_6)
assert response is None
assert not os.path.isfile(str(tmpdir.join("oct_result_000006.dump")))
response = dump_result(oct_result_10)
assert response is None
assert os.path.isfile(str(tmpdir.join("oct_result_000010.dump")))