@@ -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
298299def 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
346348def 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
359362def 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
394398def 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
409414def test_amplitude_background_redistributes_unused_budget ():
0 commit comments