doc: small refactoring on Document.add_paragraph()

Steve Canny · Steve Canny · commit 4555d2abe8bf · 2014-06-24T21:12:19.000-07:00
Document.add_paragraph() can rely on Paragraph.add_run() to add the
text.
diff --git a/docx/api.py b/docx/api.py
@@ -67,8 +67,7 @@ def add_paragraph(self, text='', style=None):
         """
         p = self._document_part.add_paragraph()
         if text:
-            r = p.add_run()
-            r.add_text(text)
+            p.add_run(text)
         if style is not None:
             p.style = style
         return p
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -70,10 +70,9 @@ def it_can_add_an_empty_paragraph(self, add_empty_paragraph_fixture):
         assert p is p_
 
     def it_can_add_a_paragraph_of_text(self, add_text_paragraph_fixture):
-        document, text, p_, r_ = add_text_paragraph_fixture
+        document, text, p_ = add_text_paragraph_fixture
         p = document.add_paragraph(text)
-        p.add_run.assert_called_once_with()
-        r_.add_text.assert_called_once_with(text)
+        p.add_run.assert_called_once_with(text)
 
     def it_can_add_a_styled_paragraph(self, add_styled_paragraph_fixture):
         document, style, p_ = add_styled_paragraph_fixture
@@ -228,9 +227,9 @@ def add_table_fixture(self, request, document, document_part_, table_):
         )
 
     @pytest.fixture
-    def add_text_paragraph_fixture(self, document, p_, r_):
+    def add_text_paragraph_fixture(self, document, p_):
         text = 'foobar\rbarfoo'
-        return document, text, p_, r_
+        return document, text, p_
 
     @pytest.fixture
     def init_fixture(self, docx_, open_):
