File size: 498 Bytes
dc17a55
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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)