Show More
@@ -61,3 +61,68 b' def test_import_pylab():' | |||||
61 | pt.import_pylab(ns, import_all=False) |
|
61 | pt.import_pylab(ns, import_all=False) | |
62 | nt.assert_true('plt' in ns) |
|
62 | nt.assert_true('plt' in ns) | |
63 | nt.assert_equal(ns['np'], np) |
|
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