PeacebinfLow commited on
Commit
dc17a55
·
verified ·
1 Parent(s): e984235

Create test_dataset_client.py

Browse files
Files changed (1) hide show
  1. 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)