Skip to content

Commit 25e971d

Browse files
committed
Update test assertions in test_google.py
Removed a redundant PartDeltaEvent assertion and updated the expected identifier in the BinaryImage snapshot assertion to 'b037a4' in the image generation test.
1 parent a68b631 commit 25e971d

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

tests/models/test_google.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1895,7 +1895,6 @@ def dummy() -> None: ... # pragma: no cover
18951895
PartDeltaEvent(index=1, delta=TextPartDelta(content_delta=IsStr())),
18961896
PartDeltaEvent(index=1, delta=TextPartDelta(content_delta=IsStr())),
18971897
PartDeltaEvent(index=1, delta=TextPartDelta(content_delta=IsStr())),
1898-
PartDeltaEvent(index=1, delta=TextPartDelta(content_delta=IsStr())),
18991898
PartEndEvent(
19001899
index=1,
19011900
part=TextPart(
@@ -3082,13 +3081,7 @@ async def test_google_vertexai_image_generation(allow_model_requests: None, vert
30823081
agent = Agent(model, output_type=BinaryImage)
30833082

30843083
result = await agent.run('Generate an image of an axolotl.')
3085-
assert result.output == snapshot(
3086-
BinaryImage(
3087-
data=IsBytes(),
3088-
media_type='image/png',
3089-
identifier='f3edd8',
3090-
)
3091-
)
3084+
assert result.output == snapshot(BinaryImage(data=IsBytes(), media_type='image/png', identifier='b037a4'))
30923085

30933086

30943087
async def test_google_httpx_client_is_not_closed(allow_model_requests: None, gemini_api_key: str):

0 commit comments

Comments
 (0)