From 14926610eeff7e201307d1919e45df074343d45e Mon Sep 17 00:00:00 2001 From: Kai Blin Date: Wed, 1 Jan 2020 18:54:03 +0100 Subject: [PATCH] core: Adjust tests for new InvalidIdError Signed-off-by: Kai Blin --- tests/test_core.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_core.py b/tests/test_core.py index 1a7da2a..7c869ac 100644 --- 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)