@@ -191,13 +191,13 @@ def test_save_tractogram_file(self):
191
191
trk_file = trk .TrkFile (tractogram )
192
192
193
193
# No need for keyword arguments.
194
- with self . assertRaises (ValueError ):
194
+ with pytest . raises (ValueError ):
195
195
nib .streamlines .save (trk_file , 'dummy.trk' , header = {})
196
196
197
197
# Wrong extension.
198
198
with pytest .warns (ExtensionWarning , match = 'extension' ):
199
199
trk_file = trk .TrkFile (tractogram )
200
- with self . assertRaises (ValueError ):
200
+ with pytest . raises (ValueError ):
201
201
nib .streamlines .save (trk_file , 'dummy.tck' , header = {})
202
202
203
203
with InTemporaryDirectory ():
@@ -272,11 +272,11 @@ def test_save_sliced_tractogram(self):
272
272
assert_tractogram_equal (tractogram , original_tractogram )
273
273
274
274
def test_load_unknown_format (self ):
275
- with self . assertRaises (ValueError ):
275
+ with pytest . raises (ValueError ):
276
276
nib .streamlines .load ('' )
277
277
278
278
def test_save_unknown_format (self ):
279
- with self . assertRaises (ValueError ):
279
+ with pytest . raises (ValueError ):
280
280
nib .streamlines .save (Tractogram (), '' )
281
281
282
282
def test_save_from_generator (self ):
0 commit comments