From 963af6a458705457019780294f3ceb8c06de9c8c Mon Sep 17 00:00:00 2001 From: smokestacklightnin <125844868+smokestacklightnin@users.noreply.github.com> Date: Tue, 4 Feb 2025 01:41:56 -0800 Subject: [PATCH] Use `zip(..., strict=True)` to force arguments to be the same length --- tests/deploy/api/test_endpoints.py | 2 +- tests/deploy/api/utils.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/deploy/api/test_endpoints.py b/tests/deploy/api/test_endpoints.py index 05ff330d..bdd6c79b 100644 --- a/tests/deploy/api/test_endpoints.py +++ b/tests/deploy/api/test_endpoints.py @@ -109,7 +109,7 @@ def test_get_document_content(tmp_local_root, mime_type): "GET", f"/api/documents/{document['id']}/content" ) as response: response_mime_type = response.headers["content-type"].split(";")[0] - received_text = response.read().decode("utf-8") + received_text = response.read().decode("utf-8").replace("\r\n", "\n") assert received_text == document_content diff --git a/tests/deploy/api/utils.py b/tests/deploy/api/utils.py index 380573f7..e34b322d 100644 --- a/tests/deploy/api/utils.py +++ b/tests/deploy/api/utils.py @@ -12,7 +12,11 @@ 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) + for document_path, mime_type in zip( + document_paths, + mime_types, + strict=True, + ) ], ) .raise_for_status()