Skip to content

Commit 5b9412e

Browse files
committed
Close controllers in resource tests
1 parent b79b9e8 commit 5b9412e

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

phy/apps/tests/test_base.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,7 @@ def recording_selector(n, cluster_ids, **kwargs):
293293

294294
# Changing either limit creates a distinct cache entry.
295295
assert calls == [1, 2, 1, 2]
296+
controller.close()
296297

297298

298299
def test_correlogram_sampling_preserves_nearby_pairs():
@@ -341,6 +342,7 @@ def capture_waveforms(spike_ids, channel_ids):
341342
eligible = subset_spikes[controller.supervisor.clustering.spike_clusters[subset_spikes] == 0]
342343
expected_indices = [0, (len(eligible) - 1) // 2, len(eligible) - 1]
343344
np.testing.assert_array_equal(selected[0], eligible[expected_indices])
345+
controller.close()
344346

345347

346348
def test_waveform_selected_clusters_share_total_budget(tempdir):
@@ -354,6 +356,7 @@ def test_waveform_selected_clusters_share_total_budget(tempdir):
354356
]
355357

356358
assert counts == [4, 3, 3]
359+
controller.close()
357360

358361

359362
def test_amplitude_background_has_stable_total_budget(tempdir):
@@ -389,6 +392,7 @@ def test_amplitude_background_has_stable_total_budget(tempdir):
389392
controller.supervisor.clustering.spikes_per_cluster[selected_cluster]
390393
)
391394
assert len(all_data[1].spike_ids) == expected_background
395+
controller.close()
392396

393397

394398
def test_amplitude_selected_clusters_share_total_budget(tempdir):
@@ -404,6 +408,7 @@ def test_amplitude_selected_clusters_share_total_budget(tempdir):
404408
assert len(selected_a.spike_ids) == 5
405409
assert len(selected_b.spike_ids) == 5
406410
assert len(background.spike_ids) == 7
411+
controller.close()
407412

408413

409414
def test_amplitude_background_redistributes_unused_budget():

0 commit comments

Comments
 (0)