chore: fix tests

Jhsmit · Feb 21, 2024 · 7fd9c58 · 7fd9c58
1 parent d476d46
commit 7fd9c58
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_hdxms_datasets.py b/tests/test_hdxms_datasets.py
@@ -86,13 +86,15 @@ def test_empty_vault(tmp_path):
     vault = DataVault(cache_dir=tmp_path)
     assert len(vault.datasets) == 0
 
-    idx = vault.remote_index
+    idx = vault.get_index()
+    assert isinstance(idx, pd.DataFrame)
     assert len(idx) > 0
 
     assert vault.fetch_dataset(DATA_ID)
     assert DATA_ID in vault.datasets
 
     ds = vault.load_dataset(DATA_ID)
+    assert isinstance(ds, DataSet)
 
     vault.clear_cache()
     assert len(vault.datasets) == 0