Skip to content

Commit 571be4e

Browse files
committed
Revert "revert test for CI"
This reverts commit c5f7702. Update uv.lock
1 parent c5f7702 commit 571be4e

File tree

2 files changed

+1573
-1867
lines changed

2 files changed

+1573
-1867
lines changed

tests/test_examples.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def test_docs_examples( # noqa: C901
118118
allow_model_requests: None,
119119
env: TestEnv,
120120
tmp_path_cwd: Path,
121+
benchmark: Any,
121122
):
122123
mocker.patch('pydantic_ai.agent.models.infer_model', side_effect=mock_infer_model)
123124
mocker.patch('pydantic_ai._utils.group_by_temporal', side_effect=mock_group_by_temporal)
@@ -222,9 +223,19 @@ def print(self, *args: Any, **kwargs: Any) -> None:
222223
test_globals: dict[str, str] = {'__name__': dunder_name}
223224

224225
if eval_example.update_examples: # pragma: lax no cover
225-
eval_example.run_print_update(example, call=call_name, module_globals=test_globals)
226+
benchmark(
227+
eval_example.run_print_update,
228+
example,
229+
call=call_name,
230+
module_globals=test_globals,
231+
)
226232
else:
227-
eval_example.run_print_check(example, call=call_name, module_globals=test_globals)
233+
benchmark(
234+
eval_example.run_print_check,
235+
example,
236+
call=call_name,
237+
module_globals=test_globals,
238+
)
228239

229240

230241
def print_callback(s: str) -> str:

0 commit comments

Comments
 (0)