Skip to content

Commit 697cd00

Browse files
Fix trailing whitespace and protected member access in tests
- Format test file to remove trailing whitespace - Restructure test_run_usage_with_request_usage to avoid accessing protected _lock member directly - Use multiple incr() calls instead to achieve same coverage without violating Pyright's reportPrivateUsage check - All tests pass and no Pyright errors
1 parent f544e0f commit 697cd00

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/test_usage_limits.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -252,28 +252,28 @@ def test_run_usage_with_request_usage():
252252
"""Test RunUsage operations with RequestUsage to ensure coverage of RequestUsage branches."""
253253
run_usage = RunUsage(requests=1, input_tokens=10, output_tokens=20, tool_calls=1)
254254
request_usage = RequestUsage(input_tokens=5, output_tokens=10)
255-
255+
256256
# Test __add__ with RequestUsage (covers line 232->236 branch)
257257
result = run_usage + request_usage
258258
assert result.requests == 2 # 1 + 1 (RequestUsage.requests property returns 1)
259259
assert result.input_tokens == 15
260260
assert result.output_tokens == 30
261261
assert result.tool_calls == 1 # RequestUsage doesn't have tool_calls
262-
262+
263263
# Test incr with RequestUsage (covers line 212->216 branch)
264264
run_usage2 = RunUsage(requests=2, input_tokens=20, output_tokens=30, tool_calls=2)
265265
run_usage2.incr(request_usage)
266266
assert run_usage2.requests == 3 # 2 + 1
267267
assert run_usage2.input_tokens == 25 # 20 + 5
268268
assert run_usage2.output_tokens == 40 # 30 + 10
269269
assert run_usage2.tool_calls == 2 # Unchanged
270-
271-
# Test accessing _lock multiple times on same instance (covers line 198->200 branch)
272-
# First access creates the lock
273-
lock1 = run_usage._lock
274-
# Second access returns the existing lock (covers the hasattr return path)
275-
lock2 = run_usage._lock
276-
assert lock1 is lock2 # Same lock instance
270+
271+
# Test multiple incr calls to ensure lock is reused (covers line 198->200 branch)
272+
# First incr creates the lock, second incr reuses it
273+
run_usage3 = RunUsage(requests=0, tool_calls=0)
274+
run_usage3.incr(request_usage) # First access creates lock
275+
run_usage3.incr(request_usage) # Second access reuses lock
276+
assert run_usage3.requests == 2
277277

278278

279279
async def test_tool_call_limit() -> None:

0 commit comments

Comments
 (0)