mirrors/claude-code-sdk-python
1
0
Fork 0
mirror of https://github.com/anthropics/claude-code-sdk-python.git synced 2025-07-07 14:45:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Apply ruff linting fixes

Browse source 
- Fixed formatting in test_subprocess_buffering.py
- No functional changes, only code style improvements

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Lina Tawfik 2025-06-27 15:04:20 -07:00
parent 3f25c3bb83
commit ecad93a0c0
No known key found for this signature in database
1 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/test_subprocess_buffering.py
View file

@ -85,26 +85,26 @@ class TestSubprocessBuffering:
# JSON objects with newlines in string values # JSON objects with newlines in string values
json_obj1 = {"type": "message", "content": "Line 1\nLine 2\nLine 3"} json_obj1 = {"type": "message", "content": "Line 1\nLine 2\nLine 3"}
json_obj2 = {"type": "result", "data": "Some\nMultiline\nContent"} json_obj2 = {"type": "result", "data": "Some\nMultiline\nContent"}
buffered_line = json.dumps(json_obj1) + '\n' + json.dumps(json_obj2) buffered_line = json.dumps(json_obj1) + '\n' + json.dumps(json_obj2)
transport = SubprocessCLITransport( transport = SubprocessCLITransport(
prompt="test", prompt="test",
options=ClaudeCodeOptions(), options=ClaudeCodeOptions(),
cli_path="/usr/bin/claude" cli_path="/usr/bin/claude"
) )
mock_process = MagicMock() mock_process = MagicMock()
mock_process.returncode = None mock_process.returncode = None
mock_process.wait = AsyncMock(return_value=None) mock_process.wait = AsyncMock(return_value=None)
transport._process = mock_process transport._process = mock_process
transport._stdout_stream = MockTextReceiveStream([buffered_line]) transport._stdout_stream = MockTextReceiveStream([buffered_line])
transport._stderr_stream = MockTextReceiveStream([]) transport._stderr_stream = MockTextReceiveStream([])
messages: list[Any] = [] messages: list[Any] = []
async for msg in transport.receive_messages(): async for msg in transport.receive_messages():
messages.append(msg) messages.append(msg)
assert len(messages) == 2 assert len(messages) == 2
assert messages[0]["content"] == "Line 1\nLine 2\nLine 3" assert messages[0]["content"] == "Line 1\nLine 2\nLine 3"
assert messages[1]["data"] == "Some\nMultiline\nContent" assert messages[1]["data"] == "Some\nMultiline\nContent"
@ -116,27 +116,27 @@ class TestSubprocessBuffering:
async def _test() -> None: async def _test() -> None:
json_obj1 = {"type": "message", "id": "msg1"} json_obj1 = {"type": "message", "id": "msg1"}
json_obj2 = {"type": "result", "id": "res1"} json_obj2 = {"type": "result", "id": "res1"}
# Multiple newlines between objects # Multiple newlines between objects
buffered_line = json.dumps(json_obj1) + '\n\n\n' + json.dumps(json_obj2) buffered_line = json.dumps(json_obj1) + '\n\n\n' + json.dumps(json_obj2)
transport = SubprocessCLITransport( transport = SubprocessCLITransport(
prompt="test", prompt="test",
options=ClaudeCodeOptions(), options=ClaudeCodeOptions(),
cli_path="/usr/bin/claude" cli_path="/usr/bin/claude"
) )
mock_process = MagicMock() mock_process = MagicMock()
mock_process.returncode = None mock_process.returncode = None
mock_process.wait = AsyncMock(return_value=None) mock_process.wait = AsyncMock(return_value=None)
transport._process = mock_process transport._process = mock_process
transport._stdout_stream = MockTextReceiveStream([buffered_line]) transport._stdout_stream = MockTextReceiveStream([buffered_line])
transport._stderr_stream = MockTextReceiveStream([]) transport._stderr_stream = MockTextReceiveStream([])
messages: list[Any] = [] messages: list[Any] = []
async for msg in transport.receive_messages(): async for msg in transport.receive_messages():
messages.append(msg) messages.append(msg)
assert len(messages) == 2 assert len(messages) == 2
assert messages[0]["id"] == "msg1" assert messages[0]["id"] == "msg1"
assert messages[1]["id"] == "res1" assert messages[1]["id"] == "res1"