atst/tests/utils/test_docx.py
2019-01-02 09:36:29 -05:00

13 lines
319 B
Python

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