atst/tests/utils/test_docx.py

14 lines
345 B
Python

from io import BytesIO
from zipfile import ZipFile
from atst.utils.docx import Docx
def test_render_docx():
data = {"droid_class": "R2"}
byte_str = BytesIO()
docx_file = Docx.render(byte_str, data=data)
zip_ = ZipFile(docx_file, mode="r")
document = zip_.read(Docx.DOCUMENT_FILE)
assert b"droid_class: R2" in document