diff --git a/IPython/utils/tests/test_shimmodule.py b/IPython/utils/tests/test_shimmodule.py
index 3a15617..814e482 100644
--- a/IPython/utils/tests/test_shimmodule.py
+++ b/IPython/utils/tests/test_shimmodule.py
@@ -1,13 +1,10 @@
+import pytest
 import sys
-import warnings
 
 from IPython.utils.shimmodule import ShimWarning
 
 
 def test_shim_warning():
     sys.modules.pop('IPython.config', None)
-    with warnings.catch_warnings(record=True) as w:
-        warnings.simplefilter("always")
+    with pytest.warns(ShimWarning):
         import IPython.config
-    assert len(w) == 1
-    assert issubclass(w[-1].category, ShimWarning)