From f3926e3b92bfac24efe198615b4c810563a96559 Mon Sep 17 00:00:00 2001 From: Cayo Puigdefabregas Date: Mon, 27 Mar 2023 11:13:58 +0200 Subject: [PATCH] fix import in tests --- tests/test_documents.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_documents.py b/tests/test_documents.py index 18888ae3..2996a58d 100644 --- a/tests/test_documents.py +++ b/tests/test_documents.py @@ -4,7 +4,7 @@ from io import BytesIO, StringIO from pathlib import Path import pytest -import teal.marshmallow +from ereuse_devicehub.teal.marshmallow import ValidationError from ereuse_devicehub import auth from ereuse_devicehub.client import Client, UserClient @@ -90,7 +90,7 @@ def test_erasure_certificate_wrong_id(client: Client): client.get( res=documents.DocumentDef.t, item='erasures/this-is-not-an-id', - status=teal.marshmallow.ValidationError, + status=ValidationError, )