Skip to content

Commit e2b0a34

Browse files
authored
Merge pull request #3189 from alejoe91/fix-randomness-in-tests
Add a couple of seeds in tests
2 parents 4c473d1 + ddf2147 commit e2b0a34

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/spikeinterface/core/tests/test_generate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -547,7 +547,7 @@ def test_generate_sorting_to_inject():
547547
durations = [10.0, 20.0]
548548
sorting = generate_sorting(num_units=10, durations=durations, sampling_frequency=30000, firing_rates=1.0)
549549
injected_sorting = generate_sorting_to_inject(
550-
sorting, [int(duration * sorting.sampling_frequency) for duration in durations]
550+
sorting, [int(duration * sorting.sampling_frequency) for duration in durations], seed=2308
551551
)
552552
num_spikes = sorting.count_num_spikes_per_unit()
553553
num_injected_spikes = injected_sorting.count_num_spikes_per_unit()

src/spikeinterface/preprocessing/tests/test_silence.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def test_silence(create_cache_folder):
1616

1717
rec = generate_recording()
1818

19-
rec0 = silence_periods(rec, list_periods=[[[0, 1000], [5000, 6000]], []], mode="zeros")
19+
rec0 = silence_periods(rec, list_periods=[[[0, 1000], [5000, 6000]], []], mode="zeros", seed=2308)
2020
rec0.save(verbose=False)
2121
traces_in0 = rec0.get_traces(segment_index=0, start_frame=0, end_frame=1000)
2222
traces_in1 = rec0.get_traces(segment_index=0, start_frame=5000, end_frame=6000)
@@ -25,7 +25,7 @@ def test_silence(create_cache_folder):
2525
assert np.all(traces_in1 == 0)
2626
assert not np.all(traces_out0 == 0)
2727

28-
rec1 = silence_periods(rec, list_periods=[[[0, 1000], [5000, 6000]], []], mode="noise")
28+
rec1 = silence_periods(rec, list_periods=[[[0, 1000], [5000, 6000]], []], mode="noise", seed=2308)
2929
rec1 = rec1.save(folder=cache_folder / "rec_w_noise", verbose=False, overwrite=True)
3030
noise_levels = get_noise_levels(rec, return_scaled=False)
3131
traces_in0 = rec1.get_traces(segment_index=0, start_frame=0, end_frame=1000)

0 commit comments

Comments
 (0)