Spaces:
Sleeping
Sleeping
Create test_dataset_client.py
Browse files- tests/test_dataset_client.py +15 -0
tests/test_dataset_client.py
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os
|
| 2 |
+
from src.dataset_client import DatasetAPIClient
|
| 3 |
+
|
| 4 |
+
def test_client_init():
|
| 5 |
+
client = DatasetAPIClient(dataset="PeacebinfLow/mindseye-lab-ledger")
|
| 6 |
+
assert client.dataset
|
| 7 |
+
|
| 8 |
+
# NOTE:
|
| 9 |
+
# Network tests are optional because HF Spaces CI doesn't always run pytest.
|
| 10 |
+
# If you run locally, you can uncomment these.
|
| 11 |
+
|
| 12 |
+
# def test_list_splits_live():
|
| 13 |
+
# client = DatasetAPIClient(dataset="PeacebinfLow/mindseye-lab-ledger")
|
| 14 |
+
# splits = client.list_splits()
|
| 15 |
+
# assert isinstance(splits, list)
|