Minor changes to heading cell in nbformat.

IPython/nbformat/v2/nbbase.py

@@ -119,12 +119,14 b' def new_text_cell(cell_type, source=None, rendered=None):'
     return cell
 
 
-def new_heading_cell(source=None, level=1):
+def new_heading_cell(source=None, rendered=None, level=1):
     """Create a new section cell with a given integer level."""
     cell = NotebookNode()
     cell.cell_type = u'heading'
     if source is not None:
         cell.source = unicode(source)
+    if rendered is not None:
+        cell.rendered = unicode(rendered)
     cell.level = int(level)
     return cell
 

IPython/nbformat/v2/rwbase.py

@@ -64,9 +64,7 b' def rejoin_lines(nb):'
                         item = output.get(key, None)
                         if isinstance(item, list):
                             output[key] = u'\n'.join(item)
-            elif cell.cell_type == 'heading':
-                pass
-            else: # text cell
+            else: # text, heading cell
                 for key in ['source', 'rendered']:
                     item = cell.get(key, None)
                     if isinstance(item, list):
@@ -92,9 +90,7 b' def split_lines(nb):'
                         item = output.get(key, None)
                         if isinstance(item, basestring):
                             output[key] = item.splitlines()
-            elif cell.cell_type == 'heading':
-                pass
-            else: # text cell
+            else: # text, heading cell
                 for key in ['source', 'rendered']:
                     item = cell.get(key, None)
                     if isinstance(item, basestring):

IPython/nbformat/v2/tests/test_nbbase.py

@@ -74,10 +74,12 b' class TestCell(TestCase):'
         tc = new_heading_cell()
         self.assertEquals(tc.cell_type, u'heading')
         self.assertEquals(u'source' not in tc, True)
+        self.assertEquals(u'rendered' not in tc, True)
 
     def test_heading_cell(self):
-        tc = new_heading_cell(u'My Heading', level=2)
-        self.assertEquals(tc.source, u'My Heading')
+        tc = new_heading_cell(u'hi', u'hi', level=2)
+        self.assertEquals(tc.source, u'hi')
+        self.assertEquals(tc.rendered, u'hi')
         self.assertEquals(tc.level, 2)