Show More
@@ -1,249 +1,249 b'' | |||||
1 | """Tests for the key interactiveshell module, where the main ipython class is defined. |
|
1 | """Tests for the key interactiveshell module, where the main ipython class is defined. | |
2 | """ |
|
2 | """ | |
3 | #----------------------------------------------------------------------------- |
|
3 | #----------------------------------------------------------------------------- | |
4 | # Module imports |
|
4 | # Module imports | |
5 | #----------------------------------------------------------------------------- |
|
5 | #----------------------------------------------------------------------------- | |
6 |
|
6 | |||
7 | # third party |
|
7 | # third party | |
8 | import pytest |
|
8 | import pytest | |
9 |
|
9 | |||
10 | # our own packages |
|
10 | # our own packages | |
11 |
|
11 | |||
12 | #----------------------------------------------------------------------------- |
|
12 | #----------------------------------------------------------------------------- | |
13 | # Test functions |
|
13 | # Test functions | |
14 | #----------------------------------------------------------------------------- |
|
14 | #----------------------------------------------------------------------------- | |
15 |
|
15 | |||
16 | def test_reset(): |
|
16 | def test_reset(): | |
17 | """reset must clear most namespaces.""" |
|
17 | """reset must clear most namespaces.""" | |
18 |
|
18 | |||
19 | # Check that reset runs without error |
|
19 | # Check that reset runs without error | |
20 | ip.reset() |
|
20 | ip.reset() | |
21 |
|
21 | |||
22 | # Once we've reset it (to clear of any junk that might have been there from |
|
22 | # Once we've reset it (to clear of any junk that might have been there from | |
23 | # other tests, we can count how many variables are in the user's namespace |
|
23 | # other tests, we can count how many variables are in the user's namespace | |
24 | nvars_user_ns = len(ip.user_ns) |
|
24 | nvars_user_ns = len(ip.user_ns) | |
25 | nvars_hidden = len(ip.user_ns_hidden) |
|
25 | nvars_hidden = len(ip.user_ns_hidden) | |
26 |
|
26 | |||
27 | # Now add a few variables to user_ns, and check that reset clears them |
|
27 | # Now add a few variables to user_ns, and check that reset clears them | |
28 | ip.user_ns['x'] = 1 |
|
28 | ip.user_ns['x'] = 1 | |
29 | ip.user_ns['y'] = 1 |
|
29 | ip.user_ns['y'] = 1 | |
30 | ip.reset() |
|
30 | ip.reset() | |
31 |
|
31 | |||
32 | # Finally, check that all namespaces have only as many variables as we |
|
32 | # Finally, check that all namespaces have only as many variables as we | |
33 | # expect to find in them: |
|
33 | # expect to find in them: | |
34 | assert len(ip.user_ns) == nvars_user_ns |
|
34 | assert len(ip.user_ns) == nvars_user_ns | |
35 | assert len(ip.user_ns_hidden) == nvars_hidden |
|
35 | assert len(ip.user_ns_hidden) == nvars_hidden | |
36 |
|
36 | |||
37 |
|
37 | |||
38 | # Tests for reporting of exceptions in various modes, handling of SystemExit, |
|
38 | # Tests for reporting of exceptions in various modes, handling of SystemExit, | |
39 | # and %tb functionality. This is really a mix of testing ultraTB and interactiveshell. |
|
39 | # and %tb functionality. This is really a mix of testing ultraTB and interactiveshell. | |
40 |
|
40 | |||
41 | def doctest_tb_plain(): |
|
41 | def doctest_tb_plain(): | |
42 | """ |
|
42 | """ | |
43 | In [18]: xmode plain |
|
43 | In [18]: xmode plain | |
44 | Exception reporting mode: Plain |
|
44 | Exception reporting mode: Plain | |
45 |
|
45 | |||
46 |
|
|
46 | In [19]: run simpleerr.py | |
47 |
|
|
47 | Traceback (most recent call last): | |
48 |
|
|
48 | ...line ..., in <module> | |
49 |
|
|
49 | bar(mode) | |
50 |
|
|
50 | ...line ..., in bar | |
51 |
|
|
51 | div0() | |
52 |
|
|
52 | ...line ..., in div0 | |
53 |
|
|
53 | x/y | |
54 |
|
|
54 | ZeroDivisionError: ... | |
55 | """ |
|
55 | """ | |
56 |
|
56 | |||
57 |
|
57 | |||
58 | def doctest_tb_context(): |
|
58 | def doctest_tb_context(): | |
59 | """ |
|
59 | """ | |
60 | In [3]: xmode context |
|
60 | In [3]: xmode context | |
61 | Exception reporting mode: Context |
|
61 | Exception reporting mode: Context | |
62 |
|
62 | |||
63 |
|
|
63 | In [4]: run simpleerr.py | |
64 |
|
|
64 | --------------------------------------------------------------------------- | |
65 |
|
|
65 | ZeroDivisionError Traceback (most recent call last) | |
66 |
|
|
66 | <BLANKLINE> | |
67 |
|
|
67 | ... in <module> | |
68 |
|
|
68 | 30 except IndexError: | |
69 |
|
|
69 | 31 mode = 'div' | |
70 |
|
|
70 | ---> 33 bar(mode) | |
71 |
|
|
71 | <BLANKLINE> | |
72 |
|
|
72 | ... in bar(mode) | |
73 |
|
|
73 | 15 "bar" | |
74 |
|
|
74 | 16 if mode=='div': | |
75 |
|
|
75 | ---> 17 div0() | |
76 |
|
|
76 | 18 elif mode=='exit': | |
77 |
|
|
77 | 19 try: | |
78 |
|
|
78 | <BLANKLINE> | |
79 |
|
|
79 | ... in div0() | |
80 |
|
|
80 | 6 x = 1 | |
81 |
|
|
81 | 7 y = 0 | |
82 |
|
|
82 | ----> 8 x/y | |
83 |
|
|
83 | <BLANKLINE> | |
84 |
|
|
84 | ZeroDivisionError: ...""" | |
85 |
|
85 | |||
86 |
|
86 | |||
87 | def doctest_tb_verbose(): |
|
87 | def doctest_tb_verbose(): | |
88 | """ |
|
88 | """ | |
89 | In [5]: xmode verbose |
|
89 | In [5]: xmode verbose | |
90 | Exception reporting mode: Verbose |
|
90 | Exception reporting mode: Verbose | |
91 |
|
91 | |||
92 | In [6]: run simpleerr.py |
|
92 | In [6]: run simpleerr.py | |
93 | --------------------------------------------------------------------------- |
|
93 | --------------------------------------------------------------------------- | |
94 | ZeroDivisionError Traceback (most recent call last) |
|
94 | ZeroDivisionError Traceback (most recent call last) | |
95 | <BLANKLINE> |
|
95 | <BLANKLINE> | |
96 | ... in <module> |
|
96 | ... in <module> | |
97 | 30 except IndexError: |
|
97 | 30 except IndexError: | |
98 | 31 mode = 'div' |
|
98 | 31 mode = 'div' | |
99 | ---> 33 bar(mode) |
|
99 | ---> 33 bar(mode) | |
100 | mode = 'div' |
|
100 | mode = 'div' | |
101 | <BLANKLINE> |
|
101 | <BLANKLINE> | |
102 | ... in bar(mode='div') |
|
102 | ... in bar(mode='div') | |
103 | 15 "bar" |
|
103 | 15 "bar" | |
104 | 16 if mode=='div': |
|
104 | 16 if mode=='div': | |
105 | ---> 17 div0() |
|
105 | ---> 17 div0() | |
106 | 18 elif mode=='exit': |
|
106 | 18 elif mode=='exit': | |
107 | 19 try: |
|
107 | 19 try: | |
108 | <BLANKLINE> |
|
108 | <BLANKLINE> | |
109 | ... in div0() |
|
109 | ... in div0() | |
110 | 6 x = 1 |
|
110 | 6 x = 1 | |
111 | 7 y = 0 |
|
111 | 7 y = 0 | |
112 | ----> 8 x/y |
|
112 | ----> 8 x/y | |
113 | x = 1 |
|
113 | x = 1 | |
114 | y = 0 |
|
114 | y = 0 | |
115 | <BLANKLINE> |
|
115 | <BLANKLINE> | |
116 | ZeroDivisionError: ... |
|
116 | ZeroDivisionError: ... | |
117 | """ |
|
117 | """ | |
118 |
|
118 | |||
119 |
|
119 | |||
120 | def doctest_tb_sysexit(): |
|
120 | def doctest_tb_sysexit(): | |
121 | """ |
|
121 | """ | |
122 | In [17]: %xmode plain |
|
122 | In [17]: %xmode plain | |
123 | Exception reporting mode: Plain |
|
123 | Exception reporting mode: Plain | |
124 |
|
124 | |||
125 | In [18]: %run simpleerr.py exit |
|
125 | In [18]: %run simpleerr.py exit | |
126 | An exception has occurred, use %tb to see the full traceback. |
|
126 | An exception has occurred, use %tb to see the full traceback. | |
127 | SystemExit: (1, 'Mode = exit') |
|
127 | SystemExit: (1, 'Mode = exit') | |
128 |
|
128 | |||
129 | In [19]: %run simpleerr.py exit 2 |
|
129 | In [19]: %run simpleerr.py exit 2 | |
130 | An exception has occurred, use %tb to see the full traceback. |
|
130 | An exception has occurred, use %tb to see the full traceback. | |
131 | SystemExit: (2, 'Mode = exit') |
|
131 | SystemExit: (2, 'Mode = exit') | |
132 |
|
132 | |||
133 | In [20]: %tb |
|
133 | In [20]: %tb | |
134 | Traceback (most recent call last): |
|
134 | Traceback (most recent call last): | |
135 | File ..., in execfile |
|
135 | File ..., in execfile | |
136 | exec(compiler(f.read(), fname, 'exec'), glob, loc) |
|
136 | exec(compiler(f.read(), fname, 'exec'), glob, loc) | |
137 | File ..., in <module> |
|
137 | File ..., in <module> | |
138 | bar(mode) |
|
138 | bar(mode) | |
139 | File ..., in bar |
|
139 | File ..., in bar | |
140 | sysexit(stat, mode) |
|
140 | sysexit(stat, mode) | |
141 | File ..., in sysexit |
|
141 | File ..., in sysexit | |
142 | raise SystemExit(stat, f"Mode = {mode}") |
|
142 | raise SystemExit(stat, f"Mode = {mode}") | |
143 | SystemExit: (2, 'Mode = exit') |
|
143 | SystemExit: (2, 'Mode = exit') | |
144 |
|
144 | |||
145 | In [21]: %xmode context |
|
145 | In [21]: %xmode context | |
146 | Exception reporting mode: Context |
|
146 | Exception reporting mode: Context | |
147 |
|
147 | |||
148 | In [22]: %tb |
|
148 | In [22]: %tb | |
149 | --------------------------------------------------------------------------- |
|
149 | --------------------------------------------------------------------------- | |
150 | SystemExit Traceback (most recent call last) |
|
150 | SystemExit Traceback (most recent call last) | |
151 | File ..., in execfile(fname, glob, loc, compiler) |
|
151 | File ..., in execfile(fname, glob, loc, compiler) | |
152 | 70 with open(fname, 'rb') as f: |
|
152 | 70 with open(fname, 'rb') as f: | |
153 | 71 compiler = compiler or compile |
|
153 | 71 compiler = compiler or compile | |
154 | ---> 72 exec(compiler(f.read(), fname, 'exec'), glob, loc) |
|
154 | ---> 72 exec(compiler(f.read(), fname, 'exec'), glob, loc) | |
155 | ...<module> |
|
155 | ...<module> | |
156 | 30 except IndexError: |
|
156 | 30 except IndexError: | |
157 | 31 mode = 'div' |
|
157 | 31 mode = 'div' | |
158 | ---> 33 bar(mode) |
|
158 | ---> 33 bar(mode) | |
159 | <BLANKLINE> |
|
159 | <BLANKLINE> | |
160 | ...bar(mode) |
|
160 | ...bar(mode) | |
161 | 21 except: |
|
161 | 21 except: | |
162 | 22 stat = 1 |
|
162 | 22 stat = 1 | |
163 | ---> 23 sysexit(stat, mode) |
|
163 | ---> 23 sysexit(stat, mode) | |
164 | 24 else: |
|
164 | 24 else: | |
165 | 25 raise ValueError('Unknown mode') |
|
165 | 25 raise ValueError('Unknown mode') | |
166 | <BLANKLINE> |
|
166 | <BLANKLINE> | |
167 | ...sysexit(stat, mode) |
|
167 | ...sysexit(stat, mode) | |
168 | 10 def sysexit(stat, mode): |
|
168 | 10 def sysexit(stat, mode): | |
169 | ---> 11 raise SystemExit(stat, f"Mode = {mode}") |
|
169 | ---> 11 raise SystemExit(stat, f"Mode = {mode}") | |
170 | <BLANKLINE> |
|
170 | <BLANKLINE> | |
171 | SystemExit: (2, 'Mode = exit') |
|
171 | SystemExit: (2, 'Mode = exit') | |
172 | """ |
|
172 | """ | |
173 |
|
173 | |||
174 |
|
174 | |||
175 | def doctest_tb_sysexit_verbose(): |
|
175 | def doctest_tb_sysexit_verbose(): | |
176 | """ |
|
176 | """ | |
177 | In [18]: %run simpleerr.py exit |
|
177 | In [18]: %run simpleerr.py exit | |
178 | An exception has occurred, use %tb to see the full traceback. |
|
178 | An exception has occurred, use %tb to see the full traceback. | |
179 | SystemExit: (1, 'Mode = exit') |
|
179 | SystemExit: (1, 'Mode = exit') | |
180 |
|
180 | |||
181 | In [19]: %run simpleerr.py exit 2 |
|
181 | In [19]: %run simpleerr.py exit 2 | |
182 | An exception has occurred, use %tb to see the full traceback. |
|
182 | An exception has occurred, use %tb to see the full traceback. | |
183 | SystemExit: (2, 'Mode = exit') |
|
183 | SystemExit: (2, 'Mode = exit') | |
184 |
|
184 | |||
185 | In [23]: %xmode verbose |
|
185 | In [23]: %xmode verbose | |
186 | Exception reporting mode: Verbose |
|
186 | Exception reporting mode: Verbose | |
187 |
|
187 | |||
188 | In [24]: %tb |
|
188 | In [24]: %tb | |
189 | --------------------------------------------------------------------------- |
|
189 | --------------------------------------------------------------------------- | |
190 | SystemExit Traceback (most recent call last) |
|
190 | SystemExit Traceback (most recent call last) | |
191 | <BLANKLINE> |
|
191 | <BLANKLINE> | |
192 | ... in <module> |
|
192 | ... in <module> | |
193 | 30 except IndexError: |
|
193 | 30 except IndexError: | |
194 | 31 mode = 'div' |
|
194 | 31 mode = 'div' | |
195 | ---> 33 bar(mode) |
|
195 | ---> 33 bar(mode) | |
196 | mode = 'exit' |
|
196 | mode = 'exit' | |
197 | <BLANKLINE> |
|
197 | <BLANKLINE> | |
198 | ... in bar(mode='exit') |
|
198 | ... in bar(mode='exit') | |
199 | ... except: |
|
199 | ... except: | |
200 | ... stat = 1 |
|
200 | ... stat = 1 | |
201 | ---> ... sysexit(stat, mode) |
|
201 | ---> ... sysexit(stat, mode) | |
202 | mode = 'exit' |
|
202 | mode = 'exit' | |
203 | stat = 2 |
|
203 | stat = 2 | |
204 | ... else: |
|
204 | ... else: | |
205 | ... raise ValueError('Unknown mode') |
|
205 | ... raise ValueError('Unknown mode') | |
206 | <BLANKLINE> |
|
206 | <BLANKLINE> | |
207 | ... in sysexit(stat=2, mode='exit') |
|
207 | ... in sysexit(stat=2, mode='exit') | |
208 | 10 def sysexit(stat, mode): |
|
208 | 10 def sysexit(stat, mode): | |
209 | ---> 11 raise SystemExit(stat, f"Mode = {mode}") |
|
209 | ---> 11 raise SystemExit(stat, f"Mode = {mode}") | |
210 | stat = 2 |
|
210 | stat = 2 | |
211 | <BLANKLINE> |
|
211 | <BLANKLINE> | |
212 | SystemExit: (2, 'Mode = exit') |
|
212 | SystemExit: (2, 'Mode = exit') | |
213 | """ |
|
213 | """ | |
214 |
|
214 | |||
215 |
|
215 | |||
216 | def test_run_cell(): |
|
216 | def test_run_cell(): | |
217 | import textwrap |
|
217 | import textwrap | |
218 |
|
218 | |||
219 | ip.run_cell("a = 10\na+=1") |
|
219 | ip.run_cell("a = 10\na+=1") | |
220 | ip.run_cell("assert a == 11\nassert 1") |
|
220 | ip.run_cell("assert a == 11\nassert 1") | |
221 |
|
221 | |||
222 | assert ip.user_ns["a"] == 11 |
|
222 | assert ip.user_ns["a"] == 11 | |
223 | complex = textwrap.dedent( |
|
223 | complex = textwrap.dedent( | |
224 | """ |
|
224 | """ | |
225 | if 1: |
|
225 | if 1: | |
226 | print "hello" |
|
226 | print "hello" | |
227 | if 1: |
|
227 | if 1: | |
228 | print "world" |
|
228 | print "world" | |
229 |
|
229 | |||
230 | if 2: |
|
230 | if 2: | |
231 | print "foo" |
|
231 | print "foo" | |
232 |
|
232 | |||
233 | if 3: |
|
233 | if 3: | |
234 | print "bar" |
|
234 | print "bar" | |
235 |
|
235 | |||
236 | if 4: |
|
236 | if 4: | |
237 | print "bar" |
|
237 | print "bar" | |
238 |
|
238 | |||
239 | """) |
|
239 | """) | |
240 | # Simply verifies that this kind of input is run |
|
240 | # Simply verifies that this kind of input is run | |
241 | ip.run_cell(complex) |
|
241 | ip.run_cell(complex) | |
242 |
|
242 | |||
243 |
|
243 | |||
244 | def test_db(): |
|
244 | def test_db(): | |
245 | """Test the internal database used for variable persistence.""" |
|
245 | """Test the internal database used for variable persistence.""" | |
246 | ip.db["__unittest_"] = 12 |
|
246 | ip.db["__unittest_"] = 12 | |
247 | assert ip.db["__unittest_"] == 12 |
|
247 | assert ip.db["__unittest_"] == 12 | |
248 | del ip.db["__unittest_"] |
|
248 | del ip.db["__unittest_"] | |
249 | assert "__unittest_" not in ip.db |
|
249 | assert "__unittest_" not in ip.db |
General Comments 0
You need to be logged in to leave comments.
Login now