Skip to content

Commit bea2e41

Browse files
committed
UT fix
1 parent b3672c7 commit bea2e41

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/tests/data/test_example.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from datetime import datetime
77
from pydantic import ValidationError
88
from judgeval.data import Example
9-
9+
from judgeval.data.tool import Tool
1010

1111
def test_basic_example_creation():
1212
example = Example(
@@ -43,7 +43,7 @@ def test_full_example_creation():
4343
assert example.retrieval_context == ["retrieval1", "retrieval2"]
4444
assert example.additional_metadata == {"key": "value"}
4545
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")]
4747
assert example.name == "test example"
4848
assert example.example_id == "123"
4949
assert example.timestamp == "20240101_120000"

0 commit comments

Comments
 (0)