catching more warnings in separation test

mir-evaluation · Mar 21, 2024 · 4878992 · 4878992
1 parent d12b583
commit 4878992
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/test_separation.py b/tests/test_separation.py
@@ -113,10 +113,11 @@ def test_empty_input(metric):
         args = [np.array([]), np.array([]), 40, 20]
 
     with pytest.warns(UserWarning, match="estimated_sources is empty,"):
-        # First, test for a warning on empty audio data
-        metric(*args)
-        # And that the metric returns empty arrays
-        assert np.allclose(metric(*args), np.array([]))
+        with pytest.warns(UserWarning, match="reference_sources is empty,"):
+            # First, test for a warning on empty audio data
+            metric(*args)
+            # And that the metric returns empty arrays
+            assert np.allclose(metric(*args), np.array([]))
 
 
 @pytest.mark.parametrize(