##// END OF EJS Templates
upstream » ipython
RSS

Clone from https://github.com/ipython/ipython.git

Make import more explicit in some notbook section...
Make import more explicit in some notbook section reimport HTML and Image in some place so that each of the subsection can be ran independently Also, link to python logo from subfolder to make the example directory more packager friendly. The actual png file will be added in next commit
Ryan May - - Load All Authors

File last commit:

r8003:82a9225c
r10084:d5eec070
Show More
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')