include ConnectionError in assertion

Author: scottyhq

tests/test_client.py

@@ -1,7 +1,7 @@
 """Tests for sliderule-python."""
 
 import pytest
-from requests.exceptions import ConnectTimeout
+from requests.exceptions import ConnectTimeout, ConnectionError
 import sliderule
 
 class TestLocal:
@@ -20,6 +20,6 @@ def test_gps2utc(self):
 @pytest.mark.network
 class TestRemote:
     def test_init_badurl(self):
-        with pytest.raises(ConnectTimeout):
+        with pytest.raises( (ConnectTimeout, ConnectionError) ):
             sliderule.set_rqst_timeout((1, 60))
             sliderule.set_url('incorrect.org')

tests/test_icesat2.py

@@ -1,7 +1,7 @@
 """Tests for sliderule-python icesat2 api."""
 
 import pytest
-from requests.exceptions import ConnectTimeout
+from requests.exceptions import ConnectTimeout, ConnectionError
 import sliderule
 from sliderule import icesat2
 from pathlib import Path
@@ -40,7 +40,7 @@ def test_toregion(self):
 @pytest.mark.network
 class TestRemote:
     def test_init_badurl(self):
-        with pytest.raises(ConnectTimeout):
+        with pytest.raises( (ConnectTimeout, ConnectionError) ):
             icesat2.init('incorrect.org')
 
     def test_get_version(self):