File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 6
6
from datetime import datetime
7
7
from pydantic import ValidationError
8
8
from judgeval .data import Example
9
-
9
+ from judgeval . data . tool import Tool
10
10
11
11
def test_basic_example_creation ():
12
12
example = Example (
@@ -43,7 +43,7 @@ def test_full_example_creation():
43
43
assert example .retrieval_context == ["retrieval1" , "retrieval2" ]
44
44
assert example .additional_metadata == {"key" : "value" }
45
45
assert example .tools_called == ["tool1" , "tool2" ]
46
- assert example .expected_tools == [{ " tool_name" : " expected_tool1"}, { " tool_name" : " expected_tool2"} ]
46
+ assert example .expected_tools == [Tool ( tool_name = " expected_tool1"), Tool ( tool_name = " expected_tool2") ]
47
47
assert example .name == "test example"
48
48
assert example .example_id == "123"
49
49
assert example .timestamp == "20240101_120000"
You can’t perform that action at this time.
0 commit comments