needle / tests /test_validation.py
Leon4gr45's picture
Upload folder using huggingface_hub
f049d1e verified
Raw History Blame Contribute Delete
963 Bytes
import pytest
from registry import normalize_tool
def test_normalization_native():
raw = {"name": "test", "description": "desc", "parameters": {"type": "object"}}
normalized = normalize_tool(raw)
assert normalized["name"] == "test"
assert normalized["description"] == "desc"
assert "parameters" in normalized
def test_normalization_openai():
raw = {
"type": "function",
"function": {
"name": "test_fn",
"description": "fn desc",
"parameters": {"type": "object"}
}
}
normalized = normalize_tool(raw)
assert normalized["name"] == "test_fn"
assert normalized["description"] == "fn desc"
assert "parameters" in normalized
def test_normalization_optional_params():
# parameters are optional and fallback to object
raw = {"name": "test", "description": "desc"}
normalized = normalize_tool(raw)
assert normalized["parameters"]["type"] == "object"