upstream/ipython Files · IPython/core/tests/test_pylabtools.py

ensure a fallback exists, so use local std{in,out,err}...

ensure a fallback exists, so use local std{in,out,err} Since IOStream instances require a valid fallback stream, use the locally defined std{in,out,err} instead of sys.std{in,out,err} in IOTerm's __init__ method. Note that the local std{in,out,err} are IOStream instances as well, that fall back to os.devnull

Fernando Perez - - Load All Authors

File last commit:

r5469:cfd87e9b


                r6652:183d9879

Download file

             test_pylabtools.py
        
                    63 lines
            
             | 1.9 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / core / tests / test_pylabtools.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      """Tests for pylab tools module.

      """

      #-----------------------------------------------------------------------------

      # Copyright (c) 2011, the IPython Development Team.

      #

      # Distributed under the terms of the Modified BSD License.

      #

      # The full license is in the file COPYING.txt, distributed with this software.

      #-----------------------------------------------------------------------------

      #-----------------------------------------------------------------------------

      # Imports

      #-----------------------------------------------------------------------------

      from __future__ import print_function

      # Stdlib imports

      # Third-party imports

      import matplotlib; matplotlib.use('Agg')

      import nose.tools as nt

      from matplotlib import pyplot as plt

      import numpy as np

      # Our own imports

      from IPython.testing import decorators as dec

      from .. import pylabtools as pt

      #-----------------------------------------------------------------------------

      # Globals and constants

      #-----------------------------------------------------------------------------

      #-----------------------------------------------------------------------------

      # Local utilities

      #-----------------------------------------------------------------------------

      #-----------------------------------------------------------------------------

      # Classes and functions

      #-----------------------------------------------------------------------------

      @dec.parametric

      def test_figure_to_svg():

          # simple empty-figure test

          fig = plt.figure()

          yield nt.assert_equal(pt.print_figure(fig, 'svg'), None)

          plt.close('all')

          # simple check for at least svg-looking output

          fig = plt.figure()

          ax = fig.add_subplot(1,1,1)

          ax.plot([1,2,3])

          plt.draw()

          svg = pt.print_figure(fig, 'svg')[:100].lower()

          yield nt.assert_true('doctype svg' in svg)

      def test_import_pylab():

          ip = get_ipython()

          ns = {}

          pt.import_pylab(ns, import_all=False)

          nt.assert_true('plt' in ns)

          nt.assert_equal(ns['np'], np)

	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

				"""Tests for pylab tools module.
				"""
				#-----------------------------------------------------------------------------
				# Copyright (c) 2011, the IPython Development Team.
				#
				# Distributed under the terms of the Modified BSD License.
				#
				# The full license is in the file COPYING.txt, distributed with this software.
				#-----------------------------------------------------------------------------

				#-----------------------------------------------------------------------------
				# Imports
				#-----------------------------------------------------------------------------
				from __future__ import print_function

				# Stdlib imports

				# Third-party imports
				import matplotlib; matplotlib.use('Agg')
				import nose.tools as nt

				from matplotlib import pyplot as plt
				import numpy as np

				# Our own imports
				from IPython.testing import decorators as dec
				from .. import pylabtools as pt

				#-----------------------------------------------------------------------------
				# Globals and constants
				#-----------------------------------------------------------------------------

				#-----------------------------------------------------------------------------
				# Local utilities
				#-----------------------------------------------------------------------------

				#-----------------------------------------------------------------------------
				# Classes and functions
				#-----------------------------------------------------------------------------

				@dec.parametric
				def test_figure_to_svg():
				# simple empty-figure test
				fig = plt.figure()
				yield nt.assert_equal(pt.print_figure(fig, 'svg'), None)

				plt.close('all')

				# simple check for at least svg-looking output
				fig = plt.figure()
				ax = fig.add_subplot(1,1,1)
				ax.plot([1,2,3])
				plt.draw()
				svg = pt.print_figure(fig, 'svg')[:100].lower()
				yield nt.assert_true('doctype svg' in svg)


				def test_import_pylab():
				ip = get_ipython()
				ns = {}
				pt.import_pylab(ns, import_all=False)
				nt.assert_true('plt' in ns)
				nt.assert_equal(ns['np'], np)