Use zip(..., strict=True) to force arguments to be the same length

Quansight · Feb 4, 2025 · 963af6a · 963af6a
1 parent 5eafb6d
commit 963af6a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git 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
@@ -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()