tests: move and update tests

qdrant · Mar 1, 2024 · c2178b3 · c2178b3
1 parent d33de18
commit c2178b3
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 70 deletions.
diff --git a/qdrant_client/async_qdrant_remote.py b/qdrant_client/async_qdrant_remote.py
@@ -119,7 +119,7 @@ def __init__(
             )
         if grpc_compression == Compression.Deflate:
             raise ValueError(
-                "grpc.Compression.Defalte is not supported. Try grpc.Compression.Gzip or grpc.Compression.NoCompression"
+                "grpc.Compression.Deflate is not supported. Try grpc.Compression.Gzip or grpc.Compression.NoCompression"
             )
         self._grpc_compression = grpc_compression
         address = f"{self._host}:{self._port}" if self._port is not None else self._host

diff --git a/qdrant_client/grpc/__init__.py b/qdrant_client/grpc/__init__.py
@@ -4,4 +4,4 @@
 from .json_with_int_pb2 import *
 from .collections_service_pb2_grpc import *
 from .points_service_pb2_grpc import *
-from .snapshots_service_pb2_grpc import *
+from .snapshots_service_pb2_grpc import *
diff --git a/qdrant_client/qdrant_remote.py b/qdrant_client/qdrant_remote.py
@@ -135,7 +135,7 @@ def __init__(
             )
         if grpc_compression == Compression.Deflate:
             raise ValueError(
-                "grpc.Compression.Defalte is not supported. Try grpc.Compression.Gzip or grpc.Compression.NoCompression"
+                "grpc.Compression.Deflate is not supported. Try grpc.Compression.Gzip or grpc.Compression.NoCompression"
             )
         self._grpc_compression = grpc_compression
 

diff --git a/tests/congruence_tests/test_grpc_channel_compress.py b/tests/congruence_tests/test_grpc_channel_compress.py
diff --git a/tests/test_qdrant_client.py b/tests/test_qdrant_client.py
@@ -7,7 +7,7 @@
 
 import numpy as np
 import pytest
-from grpc import RpcError
+from grpc import Compression, RpcError
 
 from qdrant_client import QdrantClient, models
 from qdrant_client._pydantic_compat import to_dict
@@ -48,6 +48,7 @@
 )
 from qdrant_client.qdrant_remote import QdrantRemote
 from qdrant_client.uploader.grpc_uploader import payload_to_grpc
+from tests.congruence_tests.test_common import generate_fixtures, init_client
 from tests.fixtures.payload import (
     one_random_payload_please,
     random_payload,
@@ -1730,6 +1731,21 @@ def test_grpc_options():
         )
 
 
+def test_grpc_compression():
+    client = QdrantClient(prefer_grpc=True, grpc_compression=Compression.Gzip)
+    client.get_collections()
+
+    client = QdrantClient(prefer_grpc=True, grpc_compression=Compression.NoCompression)
+    client.get_collections()
+
+    with pytest.raises(ValueError):
+        # creates a grpc client with not supported Compression type
+        QdrantClient(prefer_grpc=True, grpc_compression=Compression.Deflate)
+
+    with pytest.raises(TypeError):
+        QdrantClient(prefer_grpc=True, grpc_compression="gzip")
+
+
 if __name__ == "__main__":
     test_qdrant_client_integration()
     test_points_crud()