phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-07 11:08:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test case

Browse source
This commit is contained in:
Xi Yan 2025-03-15 16:17:22 -07:00
parent 9e6d99f7b1
commit 9da092ff2d
1 changed files with 17 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

24
tests/integration/datasets/test_datasets.py
View file

@ -16,16 +16,26 @@ import pytest
# LLAMA_STACK_CONFIG="template-name" pytest -v tests/integration/datasets
def test_register_dataset(llama_stack_client):
@pytest.mark.parametrize(
"purpose, source, provider_id",
[
(
"eval/messages-answer",
{
"type": "uri",
"uri": "huggingface://datasets/llamastack/simpleqa?split=train",
},
"huggingface",
),
],
)
def test_register_dataset(llama_stack_client, purpose, source, provider_id):
dataset = llama_stack_client.datasets.register(
purpose="eval/messages-answer",
source={
"type": "uri",
"uri": "huggingface://datasets/llamastack/simpleqa?split=train",
},
purpose=purpose,
source=source,
)
assert dataset.identifier is not None
assert dataset.provider_id == "huggingface"
assert dataset.provider_id == provider_id
iterrow_response = llama_stack_client.datasets.iterrows(
dataset.identifier, limit=10
)