assertWarns is more involved

Author: max-sixty

xarray/tests/test_extensions.py

@@ -9,7 +9,6 @@
 import xarray as xr
 
 from . import TestCase, raises_regex
-import pytest
 
 
 @xr.register_dataset_accessor('example_accessor')
@@ -53,7 +52,7 @@ def foo(self):
         del xr.Dataset.demo
         assert not hasattr(xr.Dataset, 'demo')
 
-        with pytest.warns('overriding a preexisting attribute'):
+        with self.assertWarns('overriding a preexisting attribute'):
             @xr.register_dataarray_accessor('demo')
             class Foo(object):
                 pass

xarray/tests/test_utils.py

@@ -1,3 +1,5 @@
+
+
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
@@ -18,7 +20,7 @@ def new_method():
             pass
         old_method = utils.alias(new_method, 'old_method')
         assert 'deprecated' in old_method.__doc__
-        with pytest.warns('deprecated'):
+        with self.assertWarns('deprecated'):
             old_method()
 
 

xarray/tests/test_variable.py

@@ -1697,7 +1697,7 @@ def test_concat_multiindex(self):
         assert isinstance(actual.to_index(), pd.MultiIndex)
 
     def test_coordinate_alias(self):
-        with pytest.warns('deprecated'):
+        with self.assertWarns('deprecated'):
             x = Coordinate('x', [1, 2, 3])
         assert isinstance(x, IndexVariable)