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

Moved default transformer registration into base

Ryan May - - Load All Authors

File last commit:

r8003:82a9225c


                r11429:72999967

Download file

             test_pylabtools.py
        
                    139 lines
            
             | 4.4 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / core / tests / test_pylabtools.py
          
                    History
                
                 |
                  Source
                 | Raw
                 |Copy content
                 |Copy permalink

        Fernando Perez
    
Add test file I forgot to commit.

              r3734
            
      """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

        Fernando Perez
    
Fix critical bug with pylab support inadvertently introduced in #648....

              r5468
            
      import numpy as np

        Fernando Perez
    
Add test file I forgot to commit.

              r3734
            
      # 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()

        MinRK
    
update pylabtools test to use new print_figure

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

        Fernando Perez
    
Add test file I forgot to commit.

              r3734
            
          plt.close('all')

          # simple check for at least svg-looking output

        Fernando Perez
    
Don't use new subplots() call in matplotlib so tests pass with mpl 0.99

              r3745
            
          fig = plt.figure()

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

        Fernando Perez
    
Add test file I forgot to commit.

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

          plt.draw()

        MinRK
    
update pylabtools test to use new print_figure

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

        Fernando Perez
    
Add test file I forgot to commit.

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

        Fernando Perez
    
Fix critical bug with pylab support inadvertently introduced in #648....

              r5468
            
      def test_import_pylab():

          ip = get_ipython()

        Fernando Perez
    
Refactor gui/pylab integration to eliminate code duplication....

              r5469
            
          ns = {}

          pt.import_pylab(ns, import_all=False)

          nt.assert_true('plt' in ns)

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

        Ryan May
    
Add tests of switching logic.

              r7966
            
      class TestPylabSwitch(object):

          class Shell(object):

              pylab_gui_select = None

          def setup(self):

        Ryan May
    
Improve test mocking to actually capture failure mode.

              r8003
            
              import matplotlib

              def act_mpl(backend):

                  matplotlib.rcParams['backend'] = backend

              # Save rcParams since they get modified

              self._saved_rcParams = matplotlib.rcParams

              matplotlib.rcParams = dict(backend='Qt4Agg')

              # Mock out functions

        Ryan May
    
Add tests of switching logic.

              r7966
            
              self._save_am = pt.activate_matplotlib

        Ryan May
    
Improve test mocking to actually capture failure mode.

              r8003
            
              pt.activate_matplotlib = act_mpl

        Ryan May
    
Add tests of switching logic.

              r7966
            
              self._save_ip = pt.import_pylab

              pt.import_pylab = lambda *a,**kw:None

              self._save_cis = pt.configure_inline_support

              pt.configure_inline_support = lambda *a,**kw:None

          def teardown(self):

              pt.activate_matplotlib = self._save_am

              pt.import_pylab = self._save_ip

              pt.configure_inline_support = self._save_cis

        Ryan May
    
Improve test mocking to actually capture failure mode.

              r8003
            
              import matplotlib

              matplotlib.rcParams = self._saved_rcParams

        Ryan May
    
Add tests of switching logic.

              r7966
            
          def test_qt(self):

              s = self.Shell()

        Ryan May
    
Improve test mocking to actually capture failure mode.

              r8003
            
              gui = pt.pylab_activate(dict(), None, False, s)

        Ryan May
    
Add tests of switching logic.

              r7966
            
              nt.assert_equal(gui, 'qt')

              nt.assert_equal(s.pylab_gui_select, 'qt')

              gui = pt.pylab_activate(dict(), 'inline', False, s)

              nt.assert_equal(gui, 'inline')

              nt.assert_equal(s.pylab_gui_select, 'qt')

        Ryan May
    
Improve test mocking to actually capture failure mode.

              r8003
            
              gui = pt.pylab_activate(dict(), 'qt', False, s)

        Ryan May
    
Add tests of switching logic.

              r7966
            
              nt.assert_equal(gui, 'qt')

              nt.assert_equal(s.pylab_gui_select, 'qt')

              gui = pt.pylab_activate(dict(), 'inline', False, s)

              nt.assert_equal(gui, 'inline')

              nt.assert_equal(s.pylab_gui_select, 'qt')

              gui = pt.pylab_activate(dict(), None, False, s)

              nt.assert_equal(gui, 'qt')

              nt.assert_equal(s.pylab_gui_select, 'qt')

          def test_inline(self):

              s = self.Shell()

              gui = pt.pylab_activate(dict(), 'inline', False, s)

              nt.assert_equal(gui, 'inline')

              nt.assert_equal(s.pylab_gui_select, None)

              gui = pt.pylab_activate(dict(), 'inline', False, s)

              nt.assert_equal(gui, 'inline')

              nt.assert_equal(s.pylab_gui_select, None)

              gui = pt.pylab_activate(dict(), 'qt', False, s)

              nt.assert_equal(gui, 'qt')

              nt.assert_equal(s.pylab_gui_select, 'qt')

          def test_qt_gtk(self):

              s = self.Shell()

              gui = pt.pylab_activate(dict(), 'qt', False, s)

              nt.assert_equal(gui, 'qt')

              nt.assert_equal(s.pylab_gui_select, 'qt')

              gui = pt.pylab_activate(dict(), 'gtk', False, s)

              nt.assert_equal(gui, 'qt')

              nt.assert_equal(s.pylab_gui_select, 'qt')

	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

Fernando Perez Add test file I forgot to commit.	r3734	"""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
Fernando Perez Fix critical bug with pylab support inadvertently introduced in #648....	r5468	import numpy as np
Fernando Perez Add test file I forgot to commit.	r3734
		# 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()
MinRK update pylabtools test to use new print_figure	r3984	yield nt.assert_equal(pt.print_figure(fig, 'svg'), None)
Fernando Perez Add test file I forgot to commit.	r3734
		plt.close('all')

		# simple check for at least svg-looking output
Fernando Perez Don't use new subplots() call in matplotlib so tests pass with mpl 0.99	r3745	fig = plt.figure()
		ax = fig.add_subplot(1,1,1)
Fernando Perez Add test file I forgot to commit.	r3734	ax.plot([1,2,3])
		plt.draw()
MinRK update pylabtools test to use new print_figure	r3984	svg = pt.print_figure(fig, 'svg')[:100].lower()
Fernando Perez Add test file I forgot to commit.	r3734	yield nt.assert_true('doctype svg' in svg)
Fernando Perez Fix critical bug with pylab support inadvertently introduced in #648....	r5468

		def test_import_pylab():
		ip = get_ipython()
Fernando Perez Refactor gui/pylab integration to eliminate code duplication....	r5469	ns = {}
		pt.import_pylab(ns, import_all=False)
		nt.assert_true('plt' in ns)
		nt.assert_equal(ns['np'], np)
Ryan May Add tests of switching logic.	r7966

		class TestPylabSwitch(object):
		class Shell(object):
		pylab_gui_select = None

		def setup(self):
Ryan May Improve test mocking to actually capture failure mode.	r8003	import matplotlib
		def act_mpl(backend):
		matplotlib.rcParams['backend'] = backend

		# Save rcParams since they get modified
		self._saved_rcParams = matplotlib.rcParams
		matplotlib.rcParams = dict(backend='Qt4Agg')

		# Mock out functions
Ryan May Add tests of switching logic.	r7966	self._save_am = pt.activate_matplotlib
Ryan May Improve test mocking to actually capture failure mode.	r8003	pt.activate_matplotlib = act_mpl
Ryan May Add tests of switching logic.	r7966	self._save_ip = pt.import_pylab
		pt.import_pylab = lambda a,*kw:None
		self._save_cis = pt.configure_inline_support
		pt.configure_inline_support = lambda a,*kw:None

		def teardown(self):
		pt.activate_matplotlib = self._save_am
		pt.import_pylab = self._save_ip
		pt.configure_inline_support = self._save_cis
Ryan May Improve test mocking to actually capture failure mode.	r8003	import matplotlib
		matplotlib.rcParams = self._saved_rcParams
Ryan May Add tests of switching logic.	r7966
		def test_qt(self):
		s = self.Shell()
Ryan May Improve test mocking to actually capture failure mode.	r8003	gui = pt.pylab_activate(dict(), None, False, s)
Ryan May Add tests of switching logic.	r7966	nt.assert_equal(gui, 'qt')
		nt.assert_equal(s.pylab_gui_select, 'qt')

		gui = pt.pylab_activate(dict(), 'inline', False, s)
		nt.assert_equal(gui, 'inline')
		nt.assert_equal(s.pylab_gui_select, 'qt')

Ryan May Improve test mocking to actually capture failure mode.	r8003	gui = pt.pylab_activate(dict(), 'qt', False, s)
Ryan May Add tests of switching logic.	r7966	nt.assert_equal(gui, 'qt')
		nt.assert_equal(s.pylab_gui_select, 'qt')

		gui = pt.pylab_activate(dict(), 'inline', False, s)
		nt.assert_equal(gui, 'inline')
		nt.assert_equal(s.pylab_gui_select, 'qt')

		gui = pt.pylab_activate(dict(), None, False, s)
		nt.assert_equal(gui, 'qt')
		nt.assert_equal(s.pylab_gui_select, 'qt')

		def test_inline(self):
		s = self.Shell()
		gui = pt.pylab_activate(dict(), 'inline', False, s)
		nt.assert_equal(gui, 'inline')
		nt.assert_equal(s.pylab_gui_select, None)

		gui = pt.pylab_activate(dict(), 'inline', False, s)
		nt.assert_equal(gui, 'inline')
		nt.assert_equal(s.pylab_gui_select, None)

		gui = pt.pylab_activate(dict(), 'qt', False, s)
		nt.assert_equal(gui, 'qt')
		nt.assert_equal(s.pylab_gui_select, 'qt')

		def test_qt_gtk(self):
		s = self.Shell()
		gui = pt.pylab_activate(dict(), 'qt', False, s)
		nt.assert_equal(gui, 'qt')
		nt.assert_equal(s.pylab_gui_select, 'qt')

		gui = pt.pylab_activate(dict(), 'gtk', False, s)
		nt.assert_equal(gui, 'qt')
		nt.assert_equal(s.pylab_gui_select, 'qt')