load in context manager

pydata · May 15, 2018 · 99c9661 · 99c9661
1 parent 45f0aeb
commit 99c9661
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xarray/tests/test_backends.py b/xarray/tests/test_backends.py
@@ -2654,8 +2654,8 @@ def test_pickle_rio(self):
         with create_tmp_geotiff() as (tmp_file, expected):
             with xr.open_rasterio(tmp_file) as rioda:
                 temp = pickle.dumps(rioda)
-                actual = pickle.loads(temp)
-                assert_equal(actual, rioda)
+                with pickle.loads(temp) as actual:
+                    assert_equal(actual, rioda)
 
     def test_ENVI_tags(self):
         rasterio = pytest.importorskip('rasterio', minversion='1.0a')