Skip to content

Commit ead883e

Browse files
committed
feat(add test_registry_accessibility):
1 parent 5a57b99 commit ead883e

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

tests/test_registry.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
import pytest
2+
from agentic_security.probe_data import REGISTRY
3+
from datasets import load_dataset
4+
5+
6+
@pytest.mark.slow
7+
@pytest.mark.parametrize("dataset", REGISTRY)
8+
def test_registry_accessibility(dataset):
9+
10+
source = dataset.get("source", "")
11+
if "hugging" not in source.lower():
12+
return pytest.skip("skipped dataset")
13+
14+
dataset_name = dataset.get("dataset_name")
15+
if not dataset_name:
16+
pytest.fail(f"No dataset_name found in {dataset}")
17+
18+
# Load only metadata (no data download)
19+
try:
20+
ds = load_dataset(dataset_name, split=None)
21+
# Check if metadata is accessible without loading full data
22+
assert ds is not None, f"Failed to load metadata for {dataset_name}"
23+
except Exception as e:
24+
pytest.fail(f"Error loading metadata for {dataset_name}: {str(e)}")

0 commit comments

Comments
 (0)