@@ -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