diff --git a/tests/test_OXRAPIRequest.py b/tests/test_OXRAPIRequest.py index fbfdc50ba95229c29e53918ba23f008664c2f1c4..290a8aacbc5b0e01e839ea55d1a92fe197453e1c 100644 --- a/tests/test_OXRAPIRequest.py +++ b/tests/test_OXRAPIRequest.py @@ -7,6 +7,7 @@ import string import urllib.parse import pytest +import oxrlib.errors import oxrlib.loaders from . import any_date @@ -82,11 +83,11 @@ def test_success(api_client, any_date, base): assert response.read() == (json.dumps(body) + "\n") @pytest.mark.parametrize('status_code,expect_exctype', [ - (400, oxrlib.loaders.LoaderBadRequestError), - (403, oxrlib.loaders.LoaderBadRequestError), - (404, oxrlib.loaders.LoaderNoDataError), - (410, oxrlib.loaders.LoaderNoDataError), - (500, oxrlib.loaders.LoaderSourceError), + (400, oxrlib.errors.LoaderBadRequestError), + (403, oxrlib.errors.LoaderBadRequestError), + (404, oxrlib.errors.LoaderNoDataError), + (410, oxrlib.errors.LoaderNoDataError), + (500, oxrlib.errors.LoaderSourceError), ]) def test_failure(api_client, any_date, status_code, expect_exctype): opener = FakeOpener(FakeResponse(status_code))