test_text: fixed some errors

Author: onlyjus

docx/oxml/__init__.py

@@ -63,6 +63,7 @@
 register_custom_element_class('w:dstrike', CT_OnOff)
 register_custom_element_class('w:emboss', CT_OnOff)
 register_custom_element_class('w:i', CT_OnOff)
+register_custom_element_class('w:u', CT_String)
 register_custom_element_class('w:iCs', CT_OnOff)
 register_custom_element_class('w:imprint', CT_OnOff)
 register_custom_element_class('w:noProof', CT_OnOff)

docx/oxml/table.py

@@ -297,6 +297,22 @@ def gridSpan(self):
         """
         return self.find(qn('w:gridSpan'))
 
+    def add_gridSpan(self, span):
+        """
+        Return a new <w:gridSpan> element newly inserted in sequence among
+        the existing child elements.gridSpan
+        """
+        gridSpan = CT_String.new('w:gridSpan', str(span))
+        self.append(gridSpan) # append or insert?
+        return gridSpan
+
+    @property
+    def gridSpan(self):
+        """
+        Optional <w:gridSpan> child element, or |None| if not present.
+        """
+        return self.find(qn('w:gridSpan'))
+        
 
 
 

tests/test_text.py

@@ -8,7 +8,7 @@
     absolute_import, division, print_function, unicode_literals
 )
 
-from docx.enum.text import WD_BREAK
+from docx.enum.text import WD_BREAK, WD_UNDERLINE
 from docx.oxml.text import CT_P, CT_R
 from docx.text import Paragraph, Run