Skip to content

Commit cdb15f3

Browse files
committed
Hopefully sped up tests somewhat
1 parent 3974cc5 commit cdb15f3

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

pySDC/projects/Resilience/tests/test_fault_injection.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def test_fault_injection():
153153

154154
@pytest.mark.mpi4py
155155
@pytest.mark.slow
156-
@pytest.mark.parametrize("numprocs", [5])
156+
@pytest.mark.parametrize("numprocs", [4])
157157
def test_fault_stats(numprocs):
158158
"""
159159
Test generation of fault statistics and their recovery rates
@@ -184,10 +184,10 @@ def test_fault_stats(numprocs):
184184
), f"Expected {expected_max_combinations} possible combinations for faults in van der Pol problem, but got {stats.get_max_combinations()}!"
185185

186186
recovered_reference = {
187-
'base': 1,
188-
'adaptivity': 2,
187+
'base': 0,
188+
'adaptivity': 1,
189189
'iterate': 1,
190-
'Hot Rod': 2,
190+
'Hot Rod': 1,
191191
'adaptivity_coll': 0,
192192
'double_adaptivity': 0,
193193
}
@@ -199,7 +199,7 @@ def test_fault_stats(numprocs):
199199
recovered_mask = stats.get_mask(strategy=strategy, key='recovered', op='eq', val=True)
200200
index = stats.get_index(mask=fixable_mask)
201201

202-
assert all(fixable_mask[:-1] == [False, True, False]), "Error in generating mask of fixable faults"
202+
assert all(fixable_mask == [False, True]), "Error in generating mask of fixable faults"
203203
assert all(index == [1]), "Error when converting to index"
204204

205205
combinations = np.array(stats.get_combination_counts(dat, keys=['bit'], mask=fixable_mask))
@@ -250,7 +250,7 @@ def generate_stats(load=False):
250250
],
251251
stats_path='data',
252252
)
253-
stats.run_stats_generation(runs=4, step=2)
253+
stats.run_stats_generation(runs=2, step=1)
254254
return stats
255255

256256

0 commit comments

Comments
 (0)