TEST: Small cleanups

Author: effigies

nibabel/streamlines/tests/test_streamlines.py

@@ -201,13 +201,12 @@ def test_save_tractogram_file(self):
             nib.streamlines.save(trk_file, "dummy.trk", header={})
 
         # Wrong extension.
-        with pytest.warns(None) as w:
+        with pytest.warns(ExtensionWarning) as w:
             trk_file = trk.TrkFile(tractogram)
             with self.assertRaises(ValueError):
                 nib.streamlines.save(trk_file, "dummy.tck", header={})
 
         assert len(w) == 1
-        assert issubclass(w[0].category, ExtensionWarning)
         assert "extension" in str(w[0].message)
 
         with InTemporaryDirectory():

nibabel/tests/test_image_api.py

@@ -58,7 +58,7 @@
 
 
 def maybe_deprecated(meth_name):
-    return pytest.deprecated_call() if meth_name =='get_data' else nullcontext()
+    return pytest.deprecated_call() if meth_name == 'get_data' else nullcontext()
 
 
 class GenericImageAPI(ValidateAPI):