Spaces:
Sleeping
Sleeping
File size: 466 Bytes
e272f4f |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
import pytest
from qdrant_client import QdrantClient
from qdrant_client.models import VectorParams, Distance
@pytest.fixture
def qdrant_client():
return QdrantClient(host="localhost", port=6333)
def test_collection_creation(qdrant_client):
test_collection = "test_collection"
qdrant_client.recreate_collection(test_collection, vectors_config=VectorParams(size=384, distance=Distance.COSINE))
assert qdrant_client.collection_exists(test_collection)
|