Skip to content

Commit 89e7eab

Browse files
committed
fix(tests): resolve ruff E501 line-too-long and F401 unused import
Break long test method signatures and constructor args to stay within 120-char limit. Remove unused json import.
1 parent 3098a19 commit 89e7eab

2 files changed

Lines changed: 9 additions & 4 deletions

File tree

tests/bedrock_agentcore/memory/integrations/strands/test_agentcore_memory_session_manager.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1620,7 +1620,6 @@ def test_mixed_sessions_with_blobs_and_conversational(self, batching_session_man
16201620
16211621
Note: Tests internal grouping logic by directly manipulating buffer.
16221622
"""
1623-
import json
16241623
from datetime import datetime, timezone
16251624

16261625
conv_calls = {}
@@ -1757,7 +1756,9 @@ def test_context_manager_flushes_on_exception(self, batching_session_manager, mo
17571756
assert batching_session_manager.pending_message_count() == 0
17581757
mock_memory_client.create_event.assert_called_once()
17591758

1760-
def test_exit_preserves_original_exception_when_flush_fails(self, batching_session_manager, mock_memory_client, caplog):
1759+
def test_exit_preserves_original_exception_when_flush_fails(
1760+
self, batching_session_manager, mock_memory_client, caplog
1761+
):
17611762
"""Test __exit__ logs flush failure and preserves the original exception."""
17621763
mock_memory_client.create_event.side_effect = RuntimeError("flush failed")
17631764

@@ -1777,7 +1778,9 @@ def test_exit_preserves_original_exception_when_flush_fails(self, batching_sessi
17771778
for record in caplog.records
17781779
)
17791780

1780-
def test_exit_raises_flush_exception_when_no_original_exception(self, batching_session_manager, mock_memory_client, caplog):
1781+
def test_exit_raises_flush_exception_when_no_original_exception(
1782+
self, batching_session_manager, mock_memory_client, caplog
1783+
):
17811784
"""Test __exit__ still raises flush exceptions when no original exception."""
17821785
mock_memory_client.create_event.side_effect = RuntimeError("flush failed")
17831786

tests/bedrock_agentcore/memory/integrations/strands/test_bedrock_converter.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,9 @@ def test_events_to_messages_single_event_multiple_payloads_preserves_order(self)
268268
"""Test one event with 3 conversational payloads preserves payload order."""
269269
msgs = [
270270
SessionMessage(
271-
message_id=i, message={"role": "user", "content": [{"text": f"msg{i}"}]}, created_at="2023-01-01T00:00:00Z"
271+
message_id=i,
272+
message={"role": "user", "content": [{"text": f"msg{i}"}]},
273+
created_at="2023-01-01T00:00:00Z",
272274
)
273275
for i in range(1, 4)
274276
]

0 commit comments

Comments
 (0)