mirror of
https://github.com/VectifyAI/PageIndex.git
synced 2026-04-24 23:56:21 +02:00
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
# tests/sdk/test_client.py
|
|
import pytest
|
|
from pageindex.client import PageIndexClient, LocalClient, CloudClient
|
|
|
|
|
|
def test_local_client_is_pageindex_client(tmp_path):
|
|
client = LocalClient(model="gpt-4o", storage_path=str(tmp_path / "pi"))
|
|
assert isinstance(client, PageIndexClient)
|
|
|
|
|
|
def test_cloud_client_is_pageindex_client():
|
|
client = CloudClient(api_key="pi-test")
|
|
assert isinstance(client, PageIndexClient)
|
|
|
|
|
|
def test_collection_default_name(tmp_path):
|
|
client = LocalClient(model="gpt-4o", storage_path=str(tmp_path / "pi"))
|
|
col = client.collection()
|
|
assert col.name == "default"
|
|
|
|
|
|
def test_collection_custom_name(tmp_path):
|
|
client = LocalClient(model="gpt-4o", storage_path=str(tmp_path / "pi"))
|
|
col = client.collection("papers")
|
|
assert col.name == "papers"
|
|
|
|
|
|
def test_list_collections_empty(tmp_path):
|
|
client = LocalClient(model="gpt-4o", storage_path=str(tmp_path / "pi"))
|
|
assert client.list_collections() == []
|
|
|
|
|
|
def test_list_collections_after_create(tmp_path):
|
|
client = LocalClient(model="gpt-4o", storage_path=str(tmp_path / "pi"))
|
|
client.collection("papers")
|
|
assert "papers" in client.list_collections()
|
|
|
|
|
|
def test_delete_collection(tmp_path):
|
|
client = LocalClient(model="gpt-4o", storage_path=str(tmp_path / "pi"))
|
|
client.collection("papers")
|
|
client.delete_collection("papers")
|
|
assert "papers" not in client.list_collections()
|
|
|
|
|
|
def test_register_parser(tmp_path):
|
|
client = LocalClient(model="gpt-4o", storage_path=str(tmp_path / "pi"))
|
|
class FakeParser:
|
|
def supported_extensions(self): return [".txt"]
|
|
def parse(self, file_path, **kwargs): pass
|
|
client.register_parser(FakeParser())
|