Skip to content

Commit 8ae1c91

Browse files
committed
Small updates to set random seed in generate_snippet_test.py file
1 parent 1922f84 commit 8ae1c91

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

lmfdb/tests/generate_snippet_tests.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,10 @@
4242
# To ensure output is fully deterministic and the log files don't change between runs,
4343
# we set the random seed to 1 before each test run
4444
SEED = 1
45-
seed_dict = {'sage': f'set_random_seed({SEED})',
46-
'sage_gap': f'set_random_seed({SEED})',
45+
seed_dict = {'sage': f'set_random_seed({SEED}); gap.set_seed({SEED}); libgap.set_seed({SEED})',
46+
'sage_gap': f'set_random_seed({SEED}); gap.set_seed({SEED}); libgap.set_seed({SEED})',
4747
'magma': f'SetSeed({SEED});',
48-
'oscar': f'import Random; Random.seed!({SEED}); Oscar.set_seed!({SEED});',
48+
'oscar': f'import Random; Random.seed!({SEED}); Oscar.set_seed!({SEED}); Oscar.randseed!({SEED});',
4949
'gap': f'Reset(GlobalMersenneTwister, {SEED}); Reset(GlobalRandomSource, {SEED});',
5050
'gp': f'setrand({SEED});',
5151
}
@@ -167,8 +167,9 @@ def _eval_code_file(data, lang, proc, logfile):
167167
if seed_cmd is not None:
168168
try:
169169
proc.run_command(seed_cmd, timeout=60)
170-
except Exception:
171-
print(f"Warning: could not reset random state in {lang} with {seed_cmd!r}")
170+
except Exception as exc:
171+
# Raise error if unable to set the random seed
172+
raise RuntimeError(f"Error: could not reset random state in {lang} with {seed_cmd!r}") from exc
172173

173174
with logfile.open('w') as f:
174175
proc.child.logfile = f

0 commit comments

Comments
 (0)