Skip to content

Commit 65e1848

Browse files
committed
Fix test_empty_units: -1 is a valid value for ints
1 parent 066c378 commit 65e1848

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/spikeinterface/metrics/quality/tests/test_quality_metric_calculator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,8 @@ def test_empty_units(sorting_analyzer_simple):
168168
for col in metrics_empty.columns:
169169
all_nans = np.all(isnull(metrics_empty.loc[empty_unit_ids, col].values))
170170
all_zeros = np.all(metrics_empty.loc[empty_unit_ids, col].values == 0)
171-
assert all_nans or all_zeros
171+
all_neg_ones = np.all(metrics_empty.loc[empty_unit_ids, col].values == -1)
172+
assert all_nans or all_zeros or all_neg_ones, f"Column {col} failed the empty unit test"
172173

173174

174175
if __name__ == "__main__":

0 commit comments

Comments
 (0)