parfmt: add ParagraphFormat.keep_with_next setter

Steve Canny · Steve Canny · commit 288871d6593d · 2015-02-08T00:01:44.000-08:00
diff --git a/docx/oxml/text/paragraph.py b/docx/oxml/text/paragraph.py
@@ -237,6 +237,13 @@ def keepNext_val(self):
             return None
         return keepNext.val
 
+    @keepNext_val.setter
+    def keepNext_val(self, value):
+        if value is None:
+            self._remove_keepNext()
+        else:
+            self.get_or_add_keepNext().val = value
+
     @property
     def pageBreakBefore_val(self):
         """
diff --git a/docx/text/paragraph.py b/docx/text/paragraph.py
@@ -207,6 +207,10 @@ def keep_with_next(self):
             return None
         return pPr.keepNext_val
 
+    @keep_with_next.setter
+    def keep_with_next(self, value):
+        self._element.get_or_add_pPr().keepNext_val = value
+
     @property
     def left_indent(self):
         """
diff --git a/tests/text/test_paragraph.py b/tests/text/test_paragraph.py
@@ -553,10 +553,15 @@ def on_off_get_fixture(self, request):
 
     @pytest.fixture(params=[
         ('w:p', 'keep_together',     True,  'w:p/w:pPr/w:keepLines'),
+        ('w:p', 'keep_with_next',    True,  'w:p/w:pPr/w:keepNext'),
         ('w:p/w:pPr/w:keepLines',                 'keep_together',     False,
          'w:p/w:pPr/w:keepLines{w:val=0}'),
+        ('w:p/w:pPr/w:keepNext',                  'keep_with_next',    False,
+         'w:p/w:pPr/w:keepNext{w:val=0}'),
         ('w:p/w:pPr/w:keepLines{w:val=0}',        'keep_together',     None,
          'w:p/w:pPr'),
+        ('w:p/w:pPr/w:keepNext{w:val=0}',         'keep_with_next',    None,
+         'w:p/w:pPr'),
     ])
     def on_off_set_fixture(self, request):
         p_cxml, prop_name, value, expected_cxml = request.param
