def test_getdocument(self): doc = LatexDocument() header = [LatexCommand("head"), LatexText("headtext")] content = [LatexCommand("content"), LatexText("contenttext")] doc.setHeader(header) doc.setContent(content) assert doc.getDocument() == r"""\head
def test_set_content(self): doc = LatexDocument() doc.setContent([LatexCommand("test"), LatexText("text")]) assert doc.getLinesContent()[0].getString() == r"\test" and doc.getLinesContent()[1].getString() == r"text"