phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-10 04:08:31 +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 # 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( dataset = llama_stack_client.datasets.register(
purpose="eval/messages-answer", purpose=purpose,
source={ source=source,
"type": "uri",
"uri": "huggingface://datasets/llamastack/simpleqa?split=train",
},
) )
assert dataset.identifier is not None assert dataset.identifier is not None
assert dataset.provider_id == "huggingface" assert dataset.provider_id == provider_id
iterrow_response = llama_stack_client.datasets.iterrows( iterrow_response = llama_stack_client.datasets.iterrows(
dataset.identifier, limit=10 dataset.identifier, limit=10
) )