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)