Fix bugs in SRTM tests.

SciTools · Oct 12, 2018 · c01fdb4 · c01fdb4
1 parent 6ac5d17
commit c01fdb4
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/cartopy/tests/io/test_srtm.py b/lib/cartopy/tests/io/test_srtm.py
@@ -26,6 +26,8 @@
 import cartopy.crs as ccrs
 import cartopy.io.srtm
 
+from .test_downloaders import download_to_temp
+
 
 pytestmark = [pytest.mark.skip('SRTM login not supported'),
               pytest.mark.network]
@@ -89,8 +91,10 @@ def test_out_of_range(self, Source):
     def test_in_range(self, Source):
         if Source == cartopy.io.srtm.SRTM3Source:
             shape = (1201, 1201)
-        elif Source == cartopy.io.srtm.SRTM3Source:
+        elif Source == cartopy.io.srtm.SRTM1Source:
             shape = (3601, 3601)
+        else:
+            raise ValueError('Source is of unexpected type.')
         source = Source()
         img, crs, extent = source.single_tile(-1, 50)
         assert isinstance(img, np.ndarray)