oxml: convert CT_Document to xmlchemy

Steve Canny · Steve Canny · commit e9e93a6a9276 · 2014-06-15T22:10:11.000-07:00
diff --git a/docx/oxml/parts/document.py b/docx/oxml/parts/document.py
@@ -8,16 +8,14 @@
 from ..ns import qn
 from ..table import CT_Tbl
 from ..text import CT_P
-from ..xmlchemy import BaseOxmlElement
+from ..xmlchemy import BaseOxmlElement, ZeroOrOne
 
 
 class CT_Document(BaseOxmlElement):
     """
     ``<w:document>`` element, the root element of a document.xml file.
     """
-    @property
-    def body(self):
-        return self.find(qn('w:body'))
+    body = ZeroOrOne('w:body')
 
 
 class CT_Body(BaseOxmlElement):
diff --git a/tests/oxml/parts/test_document.py b/tests/oxml/parts/test_document.py
@@ -4,10 +4,9 @@
 Test suite for the docx.oxml.parts module.
 """
 
-from docx.oxml.parts.document import CT_Body
 from docx.oxml.text import CT_P
 
-from .unitdata.document import a_body, a_document
+from .unitdata.document import a_body
 from ..unitdata.text import a_p, a_sectPr
 
 
@@ -52,10 +51,3 @@ def it_can_clear_all_the_content_it_holds(self):
             body.clear_content()
             # verify -------------------
             assert body.xml == after_body_bldr.xml()
-
-
-class DescribeCT_Document(object):
-
-    def it_holds_a_body_element(self):
-        document = a_document().with_nsdecls().with_child(a_body()).element
-        assert isinstance(document.body, CT_Body)
