Check warnings

levitsky · Dec 1, 2023 · d8095b8 · d8095b8
1 parent 351b8d7
commit d8095b8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_mgf.py b/tests/test_mgf.py
@@ -153,9 +153,10 @@ def test_write_single(self):
                 mgf.write(spectra=spectrum, output=tmpfile)
 
         self.assertGreaterEqual(len(ws), 2)
+        n_warned = 0
         for w in ws:
-            self.assertTrue(issubclass(w.category, UserWarning))
-            self.assertTrue("discouraged" in str(w.message))
+            n_warned += (issubclass(w.category, UserWarning) and "discouraged" in str(w.message))
+        self.assertGreaterEqual(n_warned, 2)
         tmpfile.seek(0)
         tmpreader = mgf.read(tmpfile)
         self.assertEqual(data.mgf_spectra_long, list(tmpreader))