upstream/ipython Files · IPython/nbformat/v1/tests/test_nbbase.py

%prun: Restore `stats.stream` after running `print_stream`....

%prun: Restore `stats.stream` after running `print_stream`. When profiling, the output from `print_stream` is captured so that the output can be paged. To do so, the `stream` attribute is set to a `StringIO` object but was not restored after the output was captured. As a result, the stats object returned by `%prun -r` did not display any output when calling its `print_stream` method. In addition, the `pstats.Stats` class has the `stream` attribute for all currently supported Python versions, so the extra branch of code can be removed.

Brian E. Granger - - Load All Authors

File last commit:

r4406:0251893c


                r7807:cb56d19d

Download file

             test_nbbase.py
        
                    41 lines
            
             | 1.1 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / nbformat / v1 / tests / test_nbbase.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      from unittest import TestCase

      from ..nbbase import (

          NotebookNode,

          new_code_cell, new_text_cell, new_notebook

      )

      class TestCell(TestCase):

          def test_empty_code_cell(self):

              cc = new_code_cell()

              self.assertEquals(cc.cell_type,'code')

              self.assertEquals('code' not in cc, True)

              self.assertEquals('prompt_number' not in cc, True)

          def test_code_cell(self):

              cc = new_code_cell(code='a=10', prompt_number=0)

              self.assertEquals(cc.code, u'a=10')

              self.assertEquals(cc.prompt_number, 0)

          def test_empty_text_cell(self):

              tc = new_text_cell()

              self.assertEquals(tc.cell_type, 'text')

              self.assertEquals('text' not in tc, True)

          def test_text_cell(self):

              tc = new_text_cell('hi')

              self.assertEquals(tc.text, u'hi')

      class TestNotebook(TestCase):

          def test_empty_notebook(self):

              nb = new_notebook()

              self.assertEquals(nb.cells, [])

          def test_notebooke(self):

              cells = [new_code_cell(),new_text_cell()]

              nb = new_notebook(cells=cells)

              self.assertEquals(nb.cells,cells)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				from unittest import TestCase

				from ..nbbase import (
				NotebookNode,
				new_code_cell, new_text_cell, new_notebook
				)

				class TestCell(TestCase):

				def test_empty_code_cell(self):
				cc = new_code_cell()
				self.assertEquals(cc.cell_type,'code')
				self.assertEquals('code' not in cc, True)
				self.assertEquals('prompt_number' not in cc, True)

				def test_code_cell(self):
				cc = new_code_cell(code='a=10', prompt_number=0)
				self.assertEquals(cc.code, u'a=10')
				self.assertEquals(cc.prompt_number, 0)

				def test_empty_text_cell(self):
				tc = new_text_cell()
				self.assertEquals(tc.cell_type, 'text')
				self.assertEquals('text' not in tc, True)

				def test_text_cell(self):
				tc = new_text_cell('hi')
				self.assertEquals(tc.text, u'hi')


				class TestNotebook(TestCase):

				def test_empty_notebook(self):
				nb = new_notebook()
				self.assertEquals(nb.cells, [])

				def test_notebooke(self):
				cells = [new_code_cell(),new_text_cell()]
				nb = new_notebook(cells=cells)
				self.assertEquals(nb.cells,cells)