remove an unused import and fix tests

astropy · Nov 9, 2022 · 4a13dd6 · 4a13dd6
1 parent 7d79d27
commit 4a13dd6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/astroquery/desi/core.py b/astroquery/desi/core.py
@@ -8,7 +8,6 @@
 import astropy.coordinates as coord
 
 from astropy import units as u
-from astropy.units import Quantity
 from astroquery.exceptions import NoResultsWarning
 from astroquery.query import BaseQuery
 from astroquery.utils import commons

diff --git a/astroquery/desi/tests/test_desi.py b/astroquery/desi/tests/test_desi.py
@@ -20,7 +20,7 @@
     'dummy_hdu_list_fits': 'hdu_list_images.fits'
 }
 
-coords = commons.ICRSCoord('11h04m27s +38d12m32s')
+coords = coord.SkyCoord('11h04m27s +38d12m32s', frame='icrs')
 radius = coord.Angle(0.5, unit='arcmin')
 pixels = 60
 data_release = 9
@@ -96,9 +96,9 @@ def compare_result_data(result, data):
 
 def image_tester(images, filetype):
     assert type(images) == list
-    data = fits.open(data_path(DATA_FILES[filetype]))
-    assert images[0][0].header == data[0].header
-    assert np.array_equal(images[0][0].data, data[0].data)
+    with fits.open(data_path(DATA_FILES[filetype])) as data:
+        assert images[0][0].header == data[0].header
+        assert np.array_equal(images[0][0].data, data[0].data)
 
 
 def test_coords_query_region(patch_tap, coords=coords, radius=radius):