tests: move and update tests

Author: joein

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

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 *

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
 

tests/congruence_tests/test_grpc_channel_compress.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,
@@ -1644,6 +1645,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()