Skip to content

Commit f50d4b4

Browse files
committed
fix coverage
1 parent 0818191 commit f50d4b4

File tree

1 file changed

+9
-16
lines changed

1 file changed

+9
-16
lines changed

tests/test_parts_manager_split_tags.py

Lines changed: 9 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -403,14 +403,11 @@ def test_start_tag_with_trailing_content_same_chunk():
403403
)
404404

405405
# Should emit event for new ThinkingPart, then delta for content
406-
assert len(events) >= 1
406+
assert len(events) == 2
407407
assert isinstance(events[0], PartStartEvent)
408408
assert isinstance(events[0].part, ThinkingPart)
409-
410-
# If content is included in the same event stream
411-
if len(events) == 2:
412-
assert isinstance(events[1], PartDeltaEvent)
413-
assert events[1].delta == ThinkingPartDelta(content_delta='thinking')
409+
assert isinstance(events[1], PartDeltaEvent)
410+
assert events[1].delta == ThinkingPartDelta(content_delta='thinking')
414411

415412
# Final state
416413
assert manager.get_parts() == snapshot([ThinkingPart(content='thinking', part_kind='thinking')])
@@ -431,13 +428,11 @@ def test_split_start_tag_with_trailing_content():
431428
)
432429

433430
# Should create ThinkingPart and add content
434-
assert len(events) >= 1
431+
assert len(events) == 2
435432
assert isinstance(events[0], PartStartEvent)
436433
assert isinstance(events[0].part, ThinkingPart)
437-
438-
if len(events) == 2:
439-
assert isinstance(events[1], PartDeltaEvent)
440-
assert events[1].delta == ThinkingPartDelta(content_delta='content')
434+
assert isinstance(events[1], PartDeltaEvent)
435+
assert events[1].delta == ThinkingPartDelta(content_delta='content')
441436

442437
assert manager.get_parts() == snapshot([ThinkingPart(content='content', part_kind='thinking')])
443438

@@ -481,13 +476,11 @@ def test_text_then_start_tag_with_content():
481476
)
482477

483478
# Should create ThinkingPart and add reasoning content
484-
assert len(events) >= 1
479+
assert len(events) == 2
485480
assert isinstance(events[0], PartStartEvent)
486481
assert isinstance(events[0].part, ThinkingPart)
487-
488-
if len(events) == 2:
489-
assert isinstance(events[1], PartDeltaEvent)
490-
assert events[1].delta == ThinkingPartDelta(content_delta='reasoning')
482+
assert isinstance(events[1], PartDeltaEvent)
483+
assert events[1].delta == ThinkingPartDelta(content_delta='reasoning')
491484

492485
# Final state
493486
assert manager.get_parts() == snapshot(

0 commit comments

Comments
 (0)