Show More
@@ -61,3 +61,68 b' def test_import_pylab():' | |||
|
61 | 61 | pt.import_pylab(ns, import_all=False) |
|
62 | 62 | nt.assert_true('plt' in ns) |
|
63 | 63 | nt.assert_equal(ns['np'], np) |
|
64 | ||
|
65 | ||
|
66 | class TestPylabSwitch(object): | |
|
67 | class Shell(object): | |
|
68 | pylab_gui_select = None | |
|
69 | ||
|
70 | def setup(self): | |
|
71 | self._save_am = pt.activate_matplotlib | |
|
72 | pt.activate_matplotlib = lambda *a,**kw:None | |
|
73 | self._save_ip = pt.import_pylab | |
|
74 | pt.import_pylab = lambda *a,**kw:None | |
|
75 | self._save_cis = pt.configure_inline_support | |
|
76 | pt.configure_inline_support = lambda *a,**kw:None | |
|
77 | ||
|
78 | def teardown(self): | |
|
79 | pt.activate_matplotlib = self._save_am | |
|
80 | pt.import_pylab = self._save_ip | |
|
81 | pt.configure_inline_support = self._save_cis | |
|
82 | ||
|
83 | def test_qt(self): | |
|
84 | s = self.Shell() | |
|
85 | gui = pt.pylab_activate(dict(), 'qt', False, s) | |
|
86 | nt.assert_equal(gui, 'qt') | |
|
87 | nt.assert_equal(s.pylab_gui_select, 'qt') | |
|
88 | ||
|
89 | gui = pt.pylab_activate(dict(), 'inline', False, s) | |
|
90 | nt.assert_equal(gui, 'inline') | |
|
91 | nt.assert_equal(s.pylab_gui_select, 'qt') | |
|
92 | ||
|
93 | gui = pt.pylab_activate(dict(), None, False, s) | |
|
94 | nt.assert_equal(gui, 'qt') | |
|
95 | nt.assert_equal(s.pylab_gui_select, 'qt') | |
|
96 | ||
|
97 | gui = pt.pylab_activate(dict(), 'inline', False, s) | |
|
98 | nt.assert_equal(gui, 'inline') | |
|
99 | nt.assert_equal(s.pylab_gui_select, 'qt') | |
|
100 | ||
|
101 | gui = pt.pylab_activate(dict(), None, False, s) | |
|
102 | nt.assert_equal(gui, 'qt') | |
|
103 | nt.assert_equal(s.pylab_gui_select, 'qt') | |
|
104 | ||
|
105 | def test_inline(self): | |
|
106 | s = self.Shell() | |
|
107 | gui = pt.pylab_activate(dict(), 'inline', False, s) | |
|
108 | nt.assert_equal(gui, 'inline') | |
|
109 | nt.assert_equal(s.pylab_gui_select, None) | |
|
110 | ||
|
111 | gui = pt.pylab_activate(dict(), 'inline', False, s) | |
|
112 | nt.assert_equal(gui, 'inline') | |
|
113 | nt.assert_equal(s.pylab_gui_select, None) | |
|
114 | ||
|
115 | gui = pt.pylab_activate(dict(), 'qt', False, s) | |
|
116 | nt.assert_equal(gui, 'qt') | |
|
117 | nt.assert_equal(s.pylab_gui_select, 'qt') | |
|
118 | ||
|
119 | def test_qt_gtk(self): | |
|
120 | s = self.Shell() | |
|
121 | gui = pt.pylab_activate(dict(), 'qt', False, s) | |
|
122 | nt.assert_equal(gui, 'qt') | |
|
123 | nt.assert_equal(s.pylab_gui_select, 'qt') | |
|
124 | ||
|
125 | gui = pt.pylab_activate(dict(), 'gtk', False, s) | |
|
126 | nt.assert_equal(gui, 'qt') | |
|
127 | nt.assert_equal(s.pylab_gui_select, 'qt') | |
|
128 |
General Comments 0
You need to be logged in to leave comments.
Login now