def test_setdocument(self): m = MeiElement("mei") doc = MeiDocument() with self.assertRaises(DocumentRootNotSetException) as cm: m.setDocument(doc) self.assertTrue(isinstance(cm.exception, DocumentRootNotSetException)) doc.setRootElement(m) self.assertEqual(doc.root, m)