def test_if_displaymath_is_set_correct_env_used(self):
doc = LaTeXDocument(r'A = \pi r^2')
doc.set_displaymath(True)
self.assertTrue('\\[' in str(doc))
self.assertTrue('\\]' in str(doc))
def test_if_displaymath_unset_correct_env_used(self):
doc = LaTeXDocument(r'A = \pi r^2')
doc.set_displaymath(False)
self.assertTrue('\\(' in str(doc))
self.assertTrue('\\)' in str(doc))
