diff --git a/tests/deploy/api/utils.py b/tests/deploy/api/utils.py index e34b322d..04352951 100644 --- a/tests/deploy/api/utils.py +++ b/tests/deploy/api/utils.py @@ -4,6 +4,8 @@ def upload_documents(*, client, document_paths, mime_types=None): if mime_types is None: mime_types = [None for _ in document_paths] + else: + assert len(mime_types) == len(document_paths) documents = ( client.post( "/api/documents", @@ -12,11 +14,7 @@ def upload_documents(*, client, document_paths, mime_types=None): "name": document_path.name, "mime_type": mime_type, } - for document_path, mime_type in zip( - document_paths, - mime_types, - strict=True, - ) + for document_path, mime_type in zip(document_paths, mime_types) ], ) .raise_for_status()