mindseye-lab-space / tests /test_dataset_client.py
PeacebinfLow's picture
Create test_dataset_client.py
dc17a55 verified
raw
history blame contribute delete
498 Bytes
import os
from src.dataset_client import DatasetAPIClient
def test_client_init():
client = DatasetAPIClient(dataset="PeacebinfLow/mindseye-lab-ledger")
assert client.dataset
# NOTE:
# Network tests are optional because HF Spaces CI doesn't always run pytest.
# If you run locally, you can uncomment these.
# def test_list_splits_live():
# client = DatasetAPIClient(dataset="PeacebinfLow/mindseye-lab-ledger")
# splits = client.list_splits()
# assert isinstance(splits, list)