improve deprecation test

Signed-off-by: xavier dupré <xavier.dupre@gmail.com>
onnx · Apr 6, 2021 · 7aff2ff · 7aff2ff
1 parent d53c2db
commit 7aff2ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_algebra_deprecation.py b/tests/test_algebra_deprecation.py
@@ -93,7 +93,8 @@ def test_sub_operator(self):
                           target_opset=TARGET_OPSET)
         mes = None
         for w in ws:
-            if w.category == DeprecationWarning:
+            if (w.category == DeprecationWarning and
+                    'numpy' not in str(w.message).lower()):
                 mes = w.message
         self.assertTrue(mes is not None)
         self.assertIn('will be removed', str(mes))