From d8095b8a16c466ad52fe54bb0c223c33dcb222b3 Mon Sep 17 00:00:00 2001 From: Joshua Klein Date: Fri, 1 Dec 2023 09:23:10 -0500 Subject: [PATCH] Check warnings --- tests/test_mgf.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_mgf.py b/tests/test_mgf.py index 3c8f942..2be3c46 100644 --- 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))