Skip to content

Commit 773d4cf

Browse files
committed
Minors in tests
1 parent 37c3749 commit 773d4cf

1 file changed

Lines changed: 4 additions & 12 deletions

File tree

tests/client/transports/test_jsonrpc_client.py

Lines changed: 4 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -485,14 +485,14 @@ async def test_send_message_streaming_comment_success(
485485
id='stream_id_123',
486486
jsonrpc='2.0',
487487
result=create_text_message_object(
488-
content='First part ', role=Role.agent
488+
content='First part', role=Role.agent
489489
),
490490
)
491491
mock_stream_response_2 = SendMessageSuccessResponse(
492492
id='stream_id_123',
493493
jsonrpc='2.0',
494494
result=create_text_message_object(
495-
content='second part ', role=Role.agent
495+
content='Second part', role=Role.agent
496496
),
497497
)
498498

@@ -521,16 +521,8 @@ async def test_send_message_streaming_comment_success(
521521
]
522522

523523
assert len(results) == 2
524-
assert isinstance(results[0], Message)
525-
assert (
526-
results[0].model_dump()
527-
== mock_stream_response_1.result.model_dump()
528-
)
529-
assert isinstance(results[1], Message)
530-
assert (
531-
results[1].model_dump()
532-
== mock_stream_response_2.result.model_dump()
533-
)
524+
assert results[0] == mock_stream_response_1.result
525+
assert results[1] == mock_stream_response_2.result
534526

535527
@pytest.mark.asyncio
536528
async def test_send_request_http_status_error(

0 commit comments

Comments
 (0)