Assert equal lengths instead of using zip(..., strict=True)

Quansight · Feb 4, 2025 · 3a2d98e · 3a2d98e
1 parent 83e7607
commit 3a2d98e
Showing 1 changed file with 3 additions and 5 deletions.
diff --git 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()