Spaces:
Running
Running
File size: 863 Bytes
c1d887c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import pytest
from backend.rag.docx_parser import parse_docx_structure
from docx import Document
from io import BytesIO
def create_mock_docx():
doc = Document()
doc.add_heading('Lesson 1: Quadratic Equations', 0)
doc.add_paragraph('Ang quadratic equation ay nasa anyong ax² + bx + c = 0.')
doc.add_heading('Objectives', level=1)
doc.add_paragraph('1. Identify quadratic terms.')
f = BytesIO()
doc.save(f)
return f.getvalue()
def test_parse_docx_structure():
content = create_mock_docx()
elements = parse_docx_structure(content)
assert len(elements) == 4
assert elements[0]['is_heading'] is True
assert 'Quadratic Equations' in elements[0]['text']
assert 'Ang quadratic equation' in elements[1]['text']
assert elements[2]['is_heading'] is True
assert elements[2]['text'] == 'Objectives'
|