| from extensions.ext_database import db | from extensions.ext_database import db | ||||
| # Check that segments were created | # Check that segments were created | ||||
| segments = db.session.query(DocumentSegment).filter_by(document_id=document.id).all() | |||||
| segments = ( | |||||
| db.session.query(DocumentSegment) | |||||
| .filter_by(document_id=document.id) | |||||
| .order_by(DocumentSegment.position) | |||||
| .all() | |||||
| ) | |||||
| assert len(segments) == 3 | assert len(segments) == 3 | ||||
| # Verify segment content and metadata | # Verify segment content and metadata |