core: Adjust tests for new InvalidIdError

Signed-off-by: Kai Blin <kblin@biosustain.dtu.dk>
kblin · Jan 1, 2020 · 1492661 · 1492661
1 parent 657d2a9
commit 1492661
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_core.py b/tests/test_core.py
@@ -13,6 +13,7 @@
 from ncbi_acc_download.errors import (
     BadPatternError,
     DownloadError,
+    InvalidIdError,
 )
 
 
@@ -223,7 +224,7 @@ def test_get_stream_bad_status(req):
     """Test getting a download stream handles bad status codes."""
     req.get(ENTREZ_URL, text=u'Nope!', status_code=404)
     params = dict(id='FAKE')
-    with pytest.raises(DownloadError):
+    with pytest.raises(InvalidIdError):
         core.get_stream(ENTREZ_URL, params)