|
@@
-1,1302
+1,1341
b''
|
|
1
|
# -*- coding: utf-8 -*-
|
|
1
|
# -*- coding: utf-8 -*-
|
|
2
|
"""Tests for various magic functions.
|
|
2
|
"""Tests for various magic functions.
|
|
3
|
|
|
3
|
|
|
4
|
Needs to be run by nose (to make ipython session available).
|
|
4
|
Needs to be run by nose (to make ipython session available).
|
|
5
|
"""
|
|
5
|
"""
|
|
6
|
|
|
6
|
|
|
|
|
|
7
|
import asyncio
|
|
7
|
import io
|
|
8
|
import io
|
|
8
|
import os
|
|
9
|
import os
|
|
9
|
import re
|
|
10
|
import re
|
|
|
|
|
11
|
import shlex
|
|
10
|
import sys
|
|
12
|
import sys
|
|
11
|
import warnings
|
|
13
|
import warnings
|
|
12
|
from textwrap import dedent
|
|
|
|
|
13
|
from unittest import TestCase
|
|
|
|
|
14
|
from unittest import mock
|
|
|
|
|
15
|
from importlib import invalidate_caches
|
|
14
|
from importlib import invalidate_caches
|
|
16
|
from io import StringIO
|
|
15
|
from io import StringIO
|
|
17
|
from pathlib import Path
|
|
16
|
from pathlib import Path
|
|
|
|
|
17
|
from textwrap import dedent
|
|
|
|
|
18
|
from unittest import TestCase, mock
|
|
18
|
|
|
19
|
|
|
19
|
import nose.tools as nt
|
|
20
|
import nose.tools as nt
|
|
20
|
|
|
21
|
|
|
21
|
import shlex
|
|
22
|
import pytest
|
|
22
|
|
|
23
|
|
|
23
|
from IPython import get_ipython
|
|
24
|
from IPython import get_ipython
|
|
24
|
from IPython.core import magic
|
|
25
|
from IPython.core import magic
|
|
25
|
from IPython.core.error import UsageError
|
|
26
|
from IPython.core.error import UsageError
|
|
26
|
from IPython.core.magic import (Magics, magics_class, line_magic,
|
|
27
|
from IPython.core.magic import (
|
|
|
|
|
28
|
Magics,
|
|
27
|
cell_magic,
|
|
29
|
cell_magic,
|
|
28
|
register_line_magic, register_cell_magic)
|
|
30
|
line_magic,
|
|
29
|
from IPython.core.magics import execution, script, code, logging, osm
|
|
31
|
magics_class,
|
|
|
|
|
32
|
register_cell_magic,
|
|
|
|
|
33
|
register_line_magic,
|
|
|
|
|
34
|
)
|
|
|
|
|
35
|
from IPython.core.magics import code, execution, logging, osm, script
|
|
30
|
from IPython.testing import decorators as dec
|
|
36
|
from IPython.testing import decorators as dec
|
|
31
|
from IPython.testing import tools as tt
|
|
37
|
from IPython.testing import tools as tt
|
|
32
|
from IPython.utils.io import capture_output
|
|
38
|
from IPython.utils.io import capture_output
|
|
33
|
from IPython.utils.tempdir import (TemporaryDirectory,
|
|
|
|
|
34
|
TemporaryWorkingDirectory)
|
|
|
|
|
35
|
from IPython.utils.process import find_cmd
|
|
39
|
from IPython.utils.process import find_cmd
|
|
36
|
from .test_debugger import PdbTestInput
|
|
40
|
from IPython.utils.tempdir import TemporaryDirectory, TemporaryWorkingDirectory
|
|
37
|
|
|
41
|
|
|
38
|
import pytest
|
|
42
|
from .test_debugger import PdbTestInput
|
|
39
|
|
|
43
|
|
|
40
|
|
|
44
|
|
|
41
|
@magic.magics_class
|
|
45
|
@magic.magics_class
|
|
42
|
class DummyMagics(magic.Magics): pass
|
|
46
|
class DummyMagics(magic.Magics): pass
|
|
43
|
|
|
47
|
|
|
44
|
def test_extract_code_ranges():
|
|
48
|
def test_extract_code_ranges():
|
|
45
|
instr = "1 3 5-6 7-9 10:15 17: :10 10- -13 :"
|
|
49
|
instr = "1 3 5-6 7-9 10:15 17: :10 10- -13 :"
|
|
46
|
expected = [(0, 1),
|
|
50
|
expected = [(0, 1),
|
|
47
|
(2, 3),
|
|
51
|
(2, 3),
|
|
48
|
(4, 6),
|
|
52
|
(4, 6),
|
|
49
|
(6, 9),
|
|
53
|
(6, 9),
|
|
50
|
(9, 14),
|
|
54
|
(9, 14),
|
|
51
|
(16, None),
|
|
55
|
(16, None),
|
|
52
|
(None, 9),
|
|
56
|
(None, 9),
|
|
53
|
(9, None),
|
|
57
|
(9, None),
|
|
54
|
(None, 13),
|
|
58
|
(None, 13),
|
|
55
|
(None, None)]
|
|
59
|
(None, None)]
|
|
56
|
actual = list(code.extract_code_ranges(instr))
|
|
60
|
actual = list(code.extract_code_ranges(instr))
|
|
57
|
nt.assert_equal(actual, expected)
|
|
61
|
nt.assert_equal(actual, expected)
|
|
58
|
|
|
62
|
|
|
59
|
def test_extract_symbols():
|
|
63
|
def test_extract_symbols():
|
|
60
|
source = """import foo\na = 10\ndef b():\n return 42\n\n\nclass A: pass\n\n\n"""
|
|
64
|
source = """import foo\na = 10\ndef b():\n return 42\n\n\nclass A: pass\n\n\n"""
|
|
61
|
symbols_args = ["a", "b", "A", "A,b", "A,a", "z"]
|
|
65
|
symbols_args = ["a", "b", "A", "A,b", "A,a", "z"]
|
|
62
|
expected = [([], ['a']),
|
|
66
|
expected = [([], ['a']),
|
|
63
|
(["def b():\n return 42\n"], []),
|
|
67
|
(["def b():\n return 42\n"], []),
|
|
64
|
(["class A: pass\n"], []),
|
|
68
|
(["class A: pass\n"], []),
|
|
65
|
(["class A: pass\n", "def b():\n return 42\n"], []),
|
|
69
|
(["class A: pass\n", "def b():\n return 42\n"], []),
|
|
66
|
(["class A: pass\n"], ['a']),
|
|
70
|
(["class A: pass\n"], ['a']),
|
|
67
|
([], ['z'])]
|
|
71
|
([], ['z'])]
|
|
68
|
for symbols, exp in zip(symbols_args, expected):
|
|
72
|
for symbols, exp in zip(symbols_args, expected):
|
|
69
|
nt.assert_equal(code.extract_symbols(source, symbols), exp)
|
|
73
|
nt.assert_equal(code.extract_symbols(source, symbols), exp)
|
|
70
|
|
|
74
|
|
|
71
|
|
|
75
|
|
|
72
|
def test_extract_symbols_raises_exception_with_non_python_code():
|
|
76
|
def test_extract_symbols_raises_exception_with_non_python_code():
|
|
73
|
source = ("=begin A Ruby program :)=end\n"
|
|
77
|
source = ("=begin A Ruby program :)=end\n"
|
|
74
|
"def hello\n"
|
|
78
|
"def hello\n"
|
|
75
|
"puts 'Hello world'\n"
|
|
79
|
"puts 'Hello world'\n"
|
|
76
|
"end")
|
|
80
|
"end")
|
|
77
|
with nt.assert_raises(SyntaxError):
|
|
81
|
with nt.assert_raises(SyntaxError):
|
|
78
|
code.extract_symbols(source, "hello")
|
|
82
|
code.extract_symbols(source, "hello")
|
|
79
|
|
|
83
|
|
|
80
|
|
|
84
|
|
|
81
|
def test_magic_not_found():
|
|
85
|
def test_magic_not_found():
|
|
82
|
# magic not found raises UsageError
|
|
86
|
# magic not found raises UsageError
|
|
83
|
with nt.assert_raises(UsageError):
|
|
87
|
with nt.assert_raises(UsageError):
|
|
84
|
_ip.magic('doesntexist')
|
|
88
|
_ip.magic('doesntexist')
|
|
85
|
|
|
89
|
|
|
86
|
# ensure result isn't success when a magic isn't found
|
|
90
|
# ensure result isn't success when a magic isn't found
|
|
87
|
result = _ip.run_cell('%doesntexist')
|
|
91
|
result = _ip.run_cell('%doesntexist')
|
|
88
|
assert isinstance(result.error_in_exec, UsageError)
|
|
92
|
assert isinstance(result.error_in_exec, UsageError)
|
|
89
|
|
|
93
|
|
|
90
|
|
|
94
|
|
|
91
|
def test_cell_magic_not_found():
|
|
95
|
def test_cell_magic_not_found():
|
|
92
|
# magic not found raises UsageError
|
|
96
|
# magic not found raises UsageError
|
|
93
|
with nt.assert_raises(UsageError):
|
|
97
|
with nt.assert_raises(UsageError):
|
|
94
|
_ip.run_cell_magic('doesntexist', 'line', 'cell')
|
|
98
|
_ip.run_cell_magic('doesntexist', 'line', 'cell')
|
|
95
|
|
|
99
|
|
|
96
|
# ensure result isn't success when a magic isn't found
|
|
100
|
# ensure result isn't success when a magic isn't found
|
|
97
|
result = _ip.run_cell('%%doesntexist')
|
|
101
|
result = _ip.run_cell('%%doesntexist')
|
|
98
|
assert isinstance(result.error_in_exec, UsageError)
|
|
102
|
assert isinstance(result.error_in_exec, UsageError)
|
|
99
|
|
|
103
|
|
|
100
|
|
|
104
|
|
|
101
|
def test_magic_error_status():
|
|
105
|
def test_magic_error_status():
|
|
102
|
def fail(shell):
|
|
106
|
def fail(shell):
|
|
103
|
1/0
|
|
107
|
1/0
|
|
104
|
_ip.register_magic_function(fail)
|
|
108
|
_ip.register_magic_function(fail)
|
|
105
|
result = _ip.run_cell('%fail')
|
|
109
|
result = _ip.run_cell('%fail')
|
|
106
|
assert isinstance(result.error_in_exec, ZeroDivisionError)
|
|
110
|
assert isinstance(result.error_in_exec, ZeroDivisionError)
|
|
107
|
|
|
111
|
|
|
108
|
|
|
112
|
|
|
109
|
def test_config():
|
|
113
|
def test_config():
|
|
110
|
""" test that config magic does not raise
|
|
114
|
""" test that config magic does not raise
|
|
111
|
can happen if Configurable init is moved too early into
|
|
115
|
can happen if Configurable init is moved too early into
|
|
112
|
Magics.__init__ as then a Config object will be registered as a
|
|
116
|
Magics.__init__ as then a Config object will be registered as a
|
|
113
|
magic.
|
|
117
|
magic.
|
|
114
|
"""
|
|
118
|
"""
|
|
115
|
## should not raise.
|
|
119
|
## should not raise.
|
|
116
|
_ip.magic('config')
|
|
120
|
_ip.magic('config')
|
|
117
|
|
|
121
|
|
|
118
|
def test_config_available_configs():
|
|
122
|
def test_config_available_configs():
|
|
119
|
""" test that config magic prints available configs in unique and
|
|
123
|
""" test that config magic prints available configs in unique and
|
|
120
|
sorted order. """
|
|
124
|
sorted order. """
|
|
121
|
with capture_output() as captured:
|
|
125
|
with capture_output() as captured:
|
|
122
|
_ip.magic('config')
|
|
126
|
_ip.magic('config')
|
|
123
|
|
|
127
|
|
|
124
|
stdout = captured.stdout
|
|
128
|
stdout = captured.stdout
|
|
125
|
config_classes = stdout.strip().split('\n')[1:]
|
|
129
|
config_classes = stdout.strip().split('\n')[1:]
|
|
126
|
nt.assert_list_equal(config_classes, sorted(set(config_classes)))
|
|
130
|
nt.assert_list_equal(config_classes, sorted(set(config_classes)))
|
|
127
|
|
|
131
|
|
|
128
|
def test_config_print_class():
|
|
132
|
def test_config_print_class():
|
|
129
|
""" test that config with a classname prints the class's options. """
|
|
133
|
""" test that config with a classname prints the class's options. """
|
|
130
|
with capture_output() as captured:
|
|
134
|
with capture_output() as captured:
|
|
131
|
_ip.magic('config TerminalInteractiveShell')
|
|
135
|
_ip.magic('config TerminalInteractiveShell')
|
|
132
|
|
|
136
|
|
|
133
|
stdout = captured.stdout
|
|
137
|
stdout = captured.stdout
|
|
134
|
if not re.match("TerminalInteractiveShell.* options", stdout.splitlines()[0]):
|
|
138
|
if not re.match("TerminalInteractiveShell.* options", stdout.splitlines()[0]):
|
|
135
|
print(stdout)
|
|
139
|
print(stdout)
|
|
136
|
raise AssertionError("1st line of stdout not like "
|
|
140
|
raise AssertionError("1st line of stdout not like "
|
|
137
|
"'TerminalInteractiveShell.* options'")
|
|
141
|
"'TerminalInteractiveShell.* options'")
|
|
138
|
|
|
142
|
|
|
139
|
def test_rehashx():
|
|
143
|
def test_rehashx():
|
|
140
|
# clear up everything
|
|
144
|
# clear up everything
|
|
141
|
_ip.alias_manager.clear_aliases()
|
|
145
|
_ip.alias_manager.clear_aliases()
|
|
142
|
del _ip.db['syscmdlist']
|
|
146
|
del _ip.db['syscmdlist']
|
|
143
|
|
|
147
|
|
|
144
|
_ip.magic('rehashx')
|
|
148
|
_ip.magic('rehashx')
|
|
145
|
# Practically ALL ipython development systems will have more than 10 aliases
|
|
149
|
# Practically ALL ipython development systems will have more than 10 aliases
|
|
146
|
|
|
150
|
|
|
147
|
nt.assert_true(len(_ip.alias_manager.aliases) > 10)
|
|
151
|
nt.assert_true(len(_ip.alias_manager.aliases) > 10)
|
|
148
|
for name, cmd in _ip.alias_manager.aliases:
|
|
152
|
for name, cmd in _ip.alias_manager.aliases:
|
|
149
|
# we must strip dots from alias names
|
|
153
|
# we must strip dots from alias names
|
|
150
|
nt.assert_not_in('.', name)
|
|
154
|
nt.assert_not_in('.', name)
|
|
151
|
|
|
155
|
|
|
152
|
# rehashx must fill up syscmdlist
|
|
156
|
# rehashx must fill up syscmdlist
|
|
153
|
scoms = _ip.db['syscmdlist']
|
|
157
|
scoms = _ip.db['syscmdlist']
|
|
154
|
nt.assert_true(len(scoms) > 10)
|
|
158
|
nt.assert_true(len(scoms) > 10)
|
|
155
|
|
|
159
|
|
|
156
|
|
|
160
|
|
|
157
|
|
|
161
|
|
|
158
|
def test_magic_parse_options():
|
|
162
|
def test_magic_parse_options():
|
|
159
|
"""Test that we don't mangle paths when parsing magic options."""
|
|
163
|
"""Test that we don't mangle paths when parsing magic options."""
|
|
160
|
ip = get_ipython()
|
|
164
|
ip = get_ipython()
|
|
161
|
path = 'c:\\x'
|
|
165
|
path = 'c:\\x'
|
|
162
|
m = DummyMagics(ip)
|
|
166
|
m = DummyMagics(ip)
|
|
163
|
opts = m.parse_options('-f %s' % path,'f:')[0]
|
|
167
|
opts = m.parse_options('-f %s' % path,'f:')[0]
|
|
164
|
# argv splitting is os-dependent
|
|
168
|
# argv splitting is os-dependent
|
|
165
|
if os.name == 'posix':
|
|
169
|
if os.name == 'posix':
|
|
166
|
expected = 'c:x'
|
|
170
|
expected = 'c:x'
|
|
167
|
else:
|
|
171
|
else:
|
|
168
|
expected = path
|
|
172
|
expected = path
|
|
169
|
nt.assert_equal(opts['f'], expected)
|
|
173
|
nt.assert_equal(opts['f'], expected)
|
|
170
|
|
|
174
|
|
|
171
|
def test_magic_parse_long_options():
|
|
175
|
def test_magic_parse_long_options():
|
|
172
|
"""Magic.parse_options can handle --foo=bar long options"""
|
|
176
|
"""Magic.parse_options can handle --foo=bar long options"""
|
|
173
|
ip = get_ipython()
|
|
177
|
ip = get_ipython()
|
|
174
|
m = DummyMagics(ip)
|
|
178
|
m = DummyMagics(ip)
|
|
175
|
opts, _ = m.parse_options('--foo --bar=bubble', 'a', 'foo', 'bar=')
|
|
179
|
opts, _ = m.parse_options('--foo --bar=bubble', 'a', 'foo', 'bar=')
|
|
176
|
nt.assert_in('foo', opts)
|
|
180
|
nt.assert_in('foo', opts)
|
|
177
|
nt.assert_in('bar', opts)
|
|
181
|
nt.assert_in('bar', opts)
|
|
178
|
nt.assert_equal(opts['bar'], "bubble")
|
|
182
|
nt.assert_equal(opts['bar'], "bubble")
|
|
179
|
|
|
183
|
|
|
180
|
|
|
184
|
|
|
181
|
def doctest_hist_f():
|
|
185
|
def doctest_hist_f():
|
|
182
|
"""Test %hist -f with temporary filename.
|
|
186
|
"""Test %hist -f with temporary filename.
|
|
183
|
|
|
187
|
|
|
184
|
In [9]: import tempfile
|
|
188
|
In [9]: import tempfile
|
|
185
|
|
|
189
|
|
|
186
|
In [10]: tfile = tempfile.mktemp('.py','tmp-ipython-')
|
|
190
|
In [10]: tfile = tempfile.mktemp('.py','tmp-ipython-')
|
|
187
|
|
|
191
|
|
|
188
|
In [11]: %hist -nl -f $tfile 3
|
|
192
|
In [11]: %hist -nl -f $tfile 3
|
|
189
|
|
|
193
|
|
|
190
|
In [13]: import os; os.unlink(tfile)
|
|
194
|
In [13]: import os; os.unlink(tfile)
|
|
191
|
"""
|
|
195
|
"""
|
|
192
|
|
|
196
|
|
|
193
|
|
|
197
|
|
|
194
|
def doctest_hist_op():
|
|
198
|
def doctest_hist_op():
|
|
195
|
"""Test %hist -op
|
|
199
|
"""Test %hist -op
|
|
196
|
|
|
200
|
|
|
197
|
In [1]: class b(float):
|
|
201
|
In [1]: class b(float):
|
|
198
|
...: pass
|
|
202
|
...: pass
|
|
199
|
...:
|
|
203
|
...:
|
|
200
|
|
|
204
|
|
|
201
|
In [2]: class s(object):
|
|
205
|
In [2]: class s(object):
|
|
202
|
...: def __str__(self):
|
|
206
|
...: def __str__(self):
|
|
203
|
...: return 's'
|
|
207
|
...: return 's'
|
|
204
|
...:
|
|
208
|
...:
|
|
205
|
|
|
209
|
|
|
206
|
In [3]:
|
|
210
|
In [3]:
|
|
207
|
|
|
211
|
|
|
208
|
In [4]: class r(b):
|
|
212
|
In [4]: class r(b):
|
|
209
|
...: def __repr__(self):
|
|
213
|
...: def __repr__(self):
|
|
210
|
...: return 'r'
|
|
214
|
...: return 'r'
|
|
211
|
...:
|
|
215
|
...:
|
|
212
|
|
|
216
|
|
|
213
|
In [5]: class sr(s,r): pass
|
|
217
|
In [5]: class sr(s,r): pass
|
|
214
|
...:
|
|
218
|
...:
|
|
215
|
|
|
219
|
|
|
216
|
In [6]:
|
|
220
|
In [6]:
|
|
217
|
|
|
221
|
|
|
218
|
In [7]: bb=b()
|
|
222
|
In [7]: bb=b()
|
|
219
|
|
|
223
|
|
|
220
|
In [8]: ss=s()
|
|
224
|
In [8]: ss=s()
|
|
221
|
|
|
225
|
|
|
222
|
In [9]: rr=r()
|
|
226
|
In [9]: rr=r()
|
|
223
|
|
|
227
|
|
|
224
|
In [10]: ssrr=sr()
|
|
228
|
In [10]: ssrr=sr()
|
|
225
|
|
|
229
|
|
|
226
|
In [11]: 4.5
|
|
230
|
In [11]: 4.5
|
|
227
|
Out[11]: 4.5
|
|
231
|
Out[11]: 4.5
|
|
228
|
|
|
232
|
|
|
229
|
In [12]: str(ss)
|
|
233
|
In [12]: str(ss)
|
|
230
|
Out[12]: 's'
|
|
234
|
Out[12]: 's'
|
|
231
|
|
|
235
|
|
|
232
|
In [13]:
|
|
236
|
In [13]:
|
|
233
|
|
|
237
|
|
|
234
|
In [14]: %hist -op
|
|
238
|
In [14]: %hist -op
|
|
235
|
>>> class b:
|
|
239
|
>>> class b:
|
|
236
|
... pass
|
|
240
|
... pass
|
|
237
|
...
|
|
241
|
...
|
|
238
|
>>> class s(b):
|
|
242
|
>>> class s(b):
|
|
239
|
... def __str__(self):
|
|
243
|
... def __str__(self):
|
|
240
|
... return 's'
|
|
244
|
... return 's'
|
|
241
|
...
|
|
245
|
...
|
|
242
|
>>>
|
|
246
|
>>>
|
|
243
|
>>> class r(b):
|
|
247
|
>>> class r(b):
|
|
244
|
... def __repr__(self):
|
|
248
|
... def __repr__(self):
|
|
245
|
... return 'r'
|
|
249
|
... return 'r'
|
|
246
|
...
|
|
250
|
...
|
|
247
|
>>> class sr(s,r): pass
|
|
251
|
>>> class sr(s,r): pass
|
|
248
|
>>>
|
|
252
|
>>>
|
|
249
|
>>> bb=b()
|
|
253
|
>>> bb=b()
|
|
250
|
>>> ss=s()
|
|
254
|
>>> ss=s()
|
|
251
|
>>> rr=r()
|
|
255
|
>>> rr=r()
|
|
252
|
>>> ssrr=sr()
|
|
256
|
>>> ssrr=sr()
|
|
253
|
>>> 4.5
|
|
257
|
>>> 4.5
|
|
254
|
4.5
|
|
258
|
4.5
|
|
255
|
>>> str(ss)
|
|
259
|
>>> str(ss)
|
|
256
|
's'
|
|
260
|
's'
|
|
257
|
>>>
|
|
261
|
>>>
|
|
258
|
"""
|
|
262
|
"""
|
|
259
|
|
|
263
|
|
|
260
|
def test_hist_pof():
|
|
264
|
def test_hist_pof():
|
|
261
|
ip = get_ipython()
|
|
265
|
ip = get_ipython()
|
|
262
|
ip.run_cell(u"1+2", store_history=True)
|
|
266
|
ip.run_cell(u"1+2", store_history=True)
|
|
263
|
#raise Exception(ip.history_manager.session_number)
|
|
267
|
#raise Exception(ip.history_manager.session_number)
|
|
264
|
#raise Exception(list(ip.history_manager._get_range_session()))
|
|
268
|
#raise Exception(list(ip.history_manager._get_range_session()))
|
|
265
|
with TemporaryDirectory() as td:
|
|
269
|
with TemporaryDirectory() as td:
|
|
266
|
tf = os.path.join(td, 'hist.py')
|
|
270
|
tf = os.path.join(td, 'hist.py')
|
|
267
|
ip.run_line_magic('history', '-pof %s' % tf)
|
|
271
|
ip.run_line_magic('history', '-pof %s' % tf)
|
|
268
|
assert os.path.isfile(tf)
|
|
272
|
assert os.path.isfile(tf)
|
|
269
|
|
|
273
|
|
|
270
|
|
|
274
|
|
|
271
|
def test_macro():
|
|
275
|
def test_macro():
|
|
272
|
ip = get_ipython()
|
|
276
|
ip = get_ipython()
|
|
273
|
ip.history_manager.reset() # Clear any existing history.
|
|
277
|
ip.history_manager.reset() # Clear any existing history.
|
|
274
|
cmds = ["a=1", "def b():\n return a**2", "print(a,b())"]
|
|
278
|
cmds = ["a=1", "def b():\n return a**2", "print(a,b())"]
|
|
275
|
for i, cmd in enumerate(cmds, start=1):
|
|
279
|
for i, cmd in enumerate(cmds, start=1):
|
|
276
|
ip.history_manager.store_inputs(i, cmd)
|
|
280
|
ip.history_manager.store_inputs(i, cmd)
|
|
277
|
ip.magic("macro test 1-3")
|
|
281
|
ip.magic("macro test 1-3")
|
|
278
|
nt.assert_equal(ip.user_ns["test"].value, "\n".join(cmds)+"\n")
|
|
282
|
nt.assert_equal(ip.user_ns["test"].value, "\n".join(cmds)+"\n")
|
|
279
|
|
|
283
|
|
|
280
|
# List macros
|
|
284
|
# List macros
|
|
281
|
nt.assert_in("test", ip.magic("macro"))
|
|
285
|
nt.assert_in("test", ip.magic("macro"))
|
|
282
|
|
|
286
|
|
|
283
|
|
|
287
|
|
|
284
|
def test_macro_run():
|
|
288
|
def test_macro_run():
|
|
285
|
"""Test that we can run a multi-line macro successfully."""
|
|
289
|
"""Test that we can run a multi-line macro successfully."""
|
|
286
|
ip = get_ipython()
|
|
290
|
ip = get_ipython()
|
|
287
|
ip.history_manager.reset()
|
|
291
|
ip.history_manager.reset()
|
|
288
|
cmds = ["a=10", "a+=1", "print(a)", "%macro test 2-3"]
|
|
292
|
cmds = ["a=10", "a+=1", "print(a)", "%macro test 2-3"]
|
|
289
|
for cmd in cmds:
|
|
293
|
for cmd in cmds:
|
|
290
|
ip.run_cell(cmd, store_history=True)
|
|
294
|
ip.run_cell(cmd, store_history=True)
|
|
291
|
nt.assert_equal(ip.user_ns["test"].value, "a+=1\nprint(a)\n")
|
|
295
|
nt.assert_equal(ip.user_ns["test"].value, "a+=1\nprint(a)\n")
|
|
292
|
with tt.AssertPrints("12"):
|
|
296
|
with tt.AssertPrints("12"):
|
|
293
|
ip.run_cell("test")
|
|
297
|
ip.run_cell("test")
|
|
294
|
with tt.AssertPrints("13"):
|
|
298
|
with tt.AssertPrints("13"):
|
|
295
|
ip.run_cell("test")
|
|
299
|
ip.run_cell("test")
|
|
296
|
|
|
300
|
|
|
297
|
|
|
301
|
|
|
298
|
def test_magic_magic():
|
|
302
|
def test_magic_magic():
|
|
299
|
"""Test %magic"""
|
|
303
|
"""Test %magic"""
|
|
300
|
ip = get_ipython()
|
|
304
|
ip = get_ipython()
|
|
301
|
with capture_output() as captured:
|
|
305
|
with capture_output() as captured:
|
|
302
|
ip.magic("magic")
|
|
306
|
ip.magic("magic")
|
|
303
|
|
|
307
|
|
|
304
|
stdout = captured.stdout
|
|
308
|
stdout = captured.stdout
|
|
305
|
nt.assert_in('%magic', stdout)
|
|
309
|
nt.assert_in('%magic', stdout)
|
|
306
|
nt.assert_in('IPython', stdout)
|
|
310
|
nt.assert_in('IPython', stdout)
|
|
307
|
nt.assert_in('Available', stdout)
|
|
311
|
nt.assert_in('Available', stdout)
|
|
308
|
|
|
312
|
|
|
309
|
|
|
313
|
|
|
310
|
@dec.skipif_not_numpy
|
|
314
|
@dec.skipif_not_numpy
|
|
311
|
def test_numpy_reset_array_undec():
|
|
315
|
def test_numpy_reset_array_undec():
|
|
312
|
"Test '%reset array' functionality"
|
|
316
|
"Test '%reset array' functionality"
|
|
313
|
_ip.ex('import numpy as np')
|
|
317
|
_ip.ex('import numpy as np')
|
|
314
|
_ip.ex('a = np.empty(2)')
|
|
318
|
_ip.ex('a = np.empty(2)')
|
|
315
|
nt.assert_in('a', _ip.user_ns)
|
|
319
|
nt.assert_in('a', _ip.user_ns)
|
|
316
|
_ip.magic('reset -f array')
|
|
320
|
_ip.magic('reset -f array')
|
|
317
|
nt.assert_not_in('a', _ip.user_ns)
|
|
321
|
nt.assert_not_in('a', _ip.user_ns)
|
|
318
|
|
|
322
|
|
|
319
|
def test_reset_out():
|
|
323
|
def test_reset_out():
|
|
320
|
"Test '%reset out' magic"
|
|
324
|
"Test '%reset out' magic"
|
|
321
|
_ip.run_cell("parrot = 'dead'", store_history=True)
|
|
325
|
_ip.run_cell("parrot = 'dead'", store_history=True)
|
|
322
|
# test '%reset -f out', make an Out prompt
|
|
326
|
# test '%reset -f out', make an Out prompt
|
|
323
|
_ip.run_cell("parrot", store_history=True)
|
|
327
|
_ip.run_cell("parrot", store_history=True)
|
|
324
|
nt.assert_true('dead' in [_ip.user_ns[x] for x in ('_','__','___')])
|
|
328
|
nt.assert_true('dead' in [_ip.user_ns[x] for x in ('_','__','___')])
|
|
325
|
_ip.magic('reset -f out')
|
|
329
|
_ip.magic('reset -f out')
|
|
326
|
nt.assert_false('dead' in [_ip.user_ns[x] for x in ('_','__','___')])
|
|
330
|
nt.assert_false('dead' in [_ip.user_ns[x] for x in ('_','__','___')])
|
|
327
|
nt.assert_equal(len(_ip.user_ns['Out']), 0)
|
|
331
|
nt.assert_equal(len(_ip.user_ns['Out']), 0)
|
|
328
|
|
|
332
|
|
|
329
|
def test_reset_in():
|
|
333
|
def test_reset_in():
|
|
330
|
"Test '%reset in' magic"
|
|
334
|
"Test '%reset in' magic"
|
|
331
|
# test '%reset -f in'
|
|
335
|
# test '%reset -f in'
|
|
332
|
_ip.run_cell("parrot", store_history=True)
|
|
336
|
_ip.run_cell("parrot", store_history=True)
|
|
333
|
nt.assert_true('parrot' in [_ip.user_ns[x] for x in ('_i','_ii','_iii')])
|
|
337
|
nt.assert_true('parrot' in [_ip.user_ns[x] for x in ('_i','_ii','_iii')])
|
|
334
|
_ip.magic('%reset -f in')
|
|
338
|
_ip.magic('%reset -f in')
|
|
335
|
nt.assert_false('parrot' in [_ip.user_ns[x] for x in ('_i','_ii','_iii')])
|
|
339
|
nt.assert_false('parrot' in [_ip.user_ns[x] for x in ('_i','_ii','_iii')])
|
|
336
|
nt.assert_equal(len(set(_ip.user_ns['In'])), 1)
|
|
340
|
nt.assert_equal(len(set(_ip.user_ns['In'])), 1)
|
|
337
|
|
|
341
|
|
|
338
|
def test_reset_dhist():
|
|
342
|
def test_reset_dhist():
|
|
339
|
"Test '%reset dhist' magic"
|
|
343
|
"Test '%reset dhist' magic"
|
|
340
|
_ip.run_cell("tmp = [d for d in _dh]") # copy before clearing
|
|
344
|
_ip.run_cell("tmp = [d for d in _dh]") # copy before clearing
|
|
341
|
_ip.magic('cd ' + os.path.dirname(nt.__file__))
|
|
345
|
_ip.magic('cd ' + os.path.dirname(nt.__file__))
|
|
342
|
_ip.magic('cd -')
|
|
346
|
_ip.magic('cd -')
|
|
343
|
nt.assert_true(len(_ip.user_ns['_dh']) > 0)
|
|
347
|
nt.assert_true(len(_ip.user_ns['_dh']) > 0)
|
|
344
|
_ip.magic('reset -f dhist')
|
|
348
|
_ip.magic('reset -f dhist')
|
|
345
|
nt.assert_equal(len(_ip.user_ns['_dh']), 0)
|
|
349
|
nt.assert_equal(len(_ip.user_ns['_dh']), 0)
|
|
346
|
_ip.run_cell("_dh = [d for d in tmp]") #restore
|
|
350
|
_ip.run_cell("_dh = [d for d in tmp]") #restore
|
|
347
|
|
|
351
|
|
|
348
|
def test_reset_in_length():
|
|
352
|
def test_reset_in_length():
|
|
349
|
"Test that '%reset in' preserves In[] length"
|
|
353
|
"Test that '%reset in' preserves In[] length"
|
|
350
|
_ip.run_cell("print 'foo'")
|
|
354
|
_ip.run_cell("print 'foo'")
|
|
351
|
_ip.run_cell("reset -f in")
|
|
355
|
_ip.run_cell("reset -f in")
|
|
352
|
nt.assert_equal(len(_ip.user_ns['In']), _ip.displayhook.prompt_count+1)
|
|
356
|
nt.assert_equal(len(_ip.user_ns['In']), _ip.displayhook.prompt_count+1)
|
|
353
|
|
|
357
|
|
|
354
|
class TestResetErrors(TestCase):
|
|
358
|
class TestResetErrors(TestCase):
|
|
355
|
|
|
359
|
|
|
356
|
def test_reset_redefine(self):
|
|
360
|
def test_reset_redefine(self):
|
|
357
|
|
|
361
|
|
|
358
|
@magics_class
|
|
362
|
@magics_class
|
|
359
|
class KernelMagics(Magics):
|
|
363
|
class KernelMagics(Magics):
|
|
360
|
@line_magic
|
|
364
|
@line_magic
|
|
361
|
def less(self, shell): pass
|
|
365
|
def less(self, shell): pass
|
|
362
|
|
|
366
|
|
|
363
|
_ip.register_magics(KernelMagics)
|
|
367
|
_ip.register_magics(KernelMagics)
|
|
364
|
|
|
368
|
|
|
365
|
with self.assertLogs() as cm:
|
|
369
|
with self.assertLogs() as cm:
|
|
366
|
# hack, we want to just capture logs, but assertLogs fails if not
|
|
370
|
# hack, we want to just capture logs, but assertLogs fails if not
|
|
367
|
# logs get produce.
|
|
371
|
# logs get produce.
|
|
368
|
# so log one things we ignore.
|
|
372
|
# so log one things we ignore.
|
|
369
|
import logging as log_mod
|
|
373
|
import logging as log_mod
|
|
370
|
log = log_mod.getLogger()
|
|
374
|
log = log_mod.getLogger()
|
|
371
|
log.info('Nothing')
|
|
375
|
log.info('Nothing')
|
|
372
|
# end hack.
|
|
376
|
# end hack.
|
|
373
|
_ip.run_cell("reset -f")
|
|
377
|
_ip.run_cell("reset -f")
|
|
374
|
|
|
378
|
|
|
375
|
assert len(cm.output) == 1
|
|
379
|
assert len(cm.output) == 1
|
|
376
|
for out in cm.output:
|
|
380
|
for out in cm.output:
|
|
377
|
assert "Invalid alias" not in out
|
|
381
|
assert "Invalid alias" not in out
|
|
378
|
|
|
382
|
|
|
379
|
def test_tb_syntaxerror():
|
|
383
|
def test_tb_syntaxerror():
|
|
380
|
"""test %tb after a SyntaxError"""
|
|
384
|
"""test %tb after a SyntaxError"""
|
|
381
|
ip = get_ipython()
|
|
385
|
ip = get_ipython()
|
|
382
|
ip.run_cell("for")
|
|
386
|
ip.run_cell("for")
|
|
383
|
|
|
387
|
|
|
384
|
# trap and validate stdout
|
|
388
|
# trap and validate stdout
|
|
385
|
save_stdout = sys.stdout
|
|
389
|
save_stdout = sys.stdout
|
|
386
|
try:
|
|
390
|
try:
|
|
387
|
sys.stdout = StringIO()
|
|
391
|
sys.stdout = StringIO()
|
|
388
|
ip.run_cell("%tb")
|
|
392
|
ip.run_cell("%tb")
|
|
389
|
out = sys.stdout.getvalue()
|
|
393
|
out = sys.stdout.getvalue()
|
|
390
|
finally:
|
|
394
|
finally:
|
|
391
|
sys.stdout = save_stdout
|
|
395
|
sys.stdout = save_stdout
|
|
392
|
# trim output, and only check the last line
|
|
396
|
# trim output, and only check the last line
|
|
393
|
last_line = out.rstrip().splitlines()[-1].strip()
|
|
397
|
last_line = out.rstrip().splitlines()[-1].strip()
|
|
394
|
nt.assert_equal(last_line, "SyntaxError: invalid syntax")
|
|
398
|
nt.assert_equal(last_line, "SyntaxError: invalid syntax")
|
|
395
|
|
|
399
|
|
|
396
|
|
|
400
|
|
|
397
|
def test_time():
|
|
401
|
def test_time():
|
|
398
|
ip = get_ipython()
|
|
402
|
ip = get_ipython()
|
|
399
|
|
|
403
|
|
|
400
|
with tt.AssertPrints("Wall time: "):
|
|
404
|
with tt.AssertPrints("Wall time: "):
|
|
401
|
ip.run_cell("%time None")
|
|
405
|
ip.run_cell("%time None")
|
|
402
|
|
|
406
|
|
|
403
|
ip.run_cell("def f(kmjy):\n"
|
|
407
|
ip.run_cell("def f(kmjy):\n"
|
|
404
|
" %time print (2*kmjy)")
|
|
408
|
" %time print (2*kmjy)")
|
|
405
|
|
|
409
|
|
|
406
|
with tt.AssertPrints("Wall time: "):
|
|
410
|
with tt.AssertPrints("Wall time: "):
|
|
407
|
with tt.AssertPrints("hihi", suppress=False):
|
|
411
|
with tt.AssertPrints("hihi", suppress=False):
|
|
408
|
ip.run_cell("f('hi')")
|
|
412
|
ip.run_cell("f('hi')")
|
|
409
|
|
|
413
|
|
|
410
|
def test_time_last_not_expression():
|
|
414
|
def test_time_last_not_expression():
|
|
411
|
ip.run_cell("%%time\n"
|
|
415
|
ip.run_cell("%%time\n"
|
|
412
|
"var_1 = 1\n"
|
|
416
|
"var_1 = 1\n"
|
|
413
|
"var_2 = 2\n")
|
|
417
|
"var_2 = 2\n")
|
|
414
|
assert ip.user_ns['var_1'] == 1
|
|
418
|
assert ip.user_ns['var_1'] == 1
|
|
415
|
del ip.user_ns['var_1']
|
|
419
|
del ip.user_ns['var_1']
|
|
416
|
assert ip.user_ns['var_2'] == 2
|
|
420
|
assert ip.user_ns['var_2'] == 2
|
|
417
|
del ip.user_ns['var_2']
|
|
421
|
del ip.user_ns['var_2']
|
|
418
|
|
|
422
|
|
|
419
|
|
|
423
|
|
|
420
|
@dec.skip_win32
|
|
424
|
@dec.skip_win32
|
|
421
|
def test_time2():
|
|
425
|
def test_time2():
|
|
422
|
ip = get_ipython()
|
|
426
|
ip = get_ipython()
|
|
423
|
|
|
427
|
|
|
424
|
with tt.AssertPrints("CPU times: user "):
|
|
428
|
with tt.AssertPrints("CPU times: user "):
|
|
425
|
ip.run_cell("%time None")
|
|
429
|
ip.run_cell("%time None")
|
|
426
|
|
|
430
|
|
|
427
|
def test_time3():
|
|
431
|
def test_time3():
|
|
428
|
"""Erroneous magic function calls, issue gh-3334"""
|
|
432
|
"""Erroneous magic function calls, issue gh-3334"""
|
|
429
|
ip = get_ipython()
|
|
433
|
ip = get_ipython()
|
|
430
|
ip.user_ns.pop('run', None)
|
|
434
|
ip.user_ns.pop('run', None)
|
|
431
|
|
|
435
|
|
|
432
|
with tt.AssertNotPrints("not found", channel='stderr'):
|
|
436
|
with tt.AssertNotPrints("not found", channel='stderr'):
|
|
433
|
ip.run_cell("%%time\n"
|
|
437
|
ip.run_cell("%%time\n"
|
|
434
|
"run = 0\n"
|
|
438
|
"run = 0\n"
|
|
435
|
"run += 1")
|
|
439
|
"run += 1")
|
|
436
|
|
|
440
|
|
|
437
|
def test_multiline_time():
|
|
441
|
def test_multiline_time():
|
|
438
|
"""Make sure last statement from time return a value."""
|
|
442
|
"""Make sure last statement from time return a value."""
|
|
439
|
ip = get_ipython()
|
|
443
|
ip = get_ipython()
|
|
440
|
ip.user_ns.pop('run', None)
|
|
444
|
ip.user_ns.pop('run', None)
|
|
441
|
|
|
445
|
|
|
442
|
ip.run_cell(dedent("""\
|
|
446
|
ip.run_cell(dedent("""\
|
|
443
|
%%time
|
|
447
|
%%time
|
|
444
|
a = "ho"
|
|
448
|
a = "ho"
|
|
445
|
b = "hey"
|
|
449
|
b = "hey"
|
|
446
|
a+b
|
|
450
|
a+b
|
|
447
|
"""))
|
|
451
|
"""))
|
|
448
|
nt.assert_equal(ip.user_ns_hidden['_'], 'hohey')
|
|
452
|
nt.assert_equal(ip.user_ns_hidden['_'], 'hohey')
|
|
449
|
|
|
453
|
|
|
450
|
def test_time_local_ns():
|
|
454
|
def test_time_local_ns():
|
|
451
|
"""
|
|
455
|
"""
|
|
452
|
Test that local_ns is actually global_ns when running a cell magic
|
|
456
|
Test that local_ns is actually global_ns when running a cell magic
|
|
453
|
"""
|
|
457
|
"""
|
|
454
|
ip = get_ipython()
|
|
458
|
ip = get_ipython()
|
|
455
|
ip.run_cell("%%time\n"
|
|
459
|
ip.run_cell("%%time\n"
|
|
456
|
"myvar = 1")
|
|
460
|
"myvar = 1")
|
|
457
|
nt.assert_equal(ip.user_ns['myvar'], 1)
|
|
461
|
nt.assert_equal(ip.user_ns['myvar'], 1)
|
|
458
|
del ip.user_ns['myvar']
|
|
462
|
del ip.user_ns['myvar']
|
|
459
|
|
|
463
|
|
|
460
|
def test_doctest_mode():
|
|
464
|
def test_doctest_mode():
|
|
461
|
"Toggle doctest_mode twice, it should be a no-op and run without error"
|
|
465
|
"Toggle doctest_mode twice, it should be a no-op and run without error"
|
|
462
|
_ip.magic('doctest_mode')
|
|
466
|
_ip.magic('doctest_mode')
|
|
463
|
_ip.magic('doctest_mode')
|
|
467
|
_ip.magic('doctest_mode')
|
|
464
|
|
|
468
|
|
|
465
|
|
|
469
|
|
|
466
|
def test_parse_options():
|
|
470
|
def test_parse_options():
|
|
467
|
"""Tests for basic options parsing in magics."""
|
|
471
|
"""Tests for basic options parsing in magics."""
|
|
468
|
# These are only the most minimal of tests, more should be added later. At
|
|
472
|
# These are only the most minimal of tests, more should be added later. At
|
|
469
|
# the very least we check that basic text/unicode calls work OK.
|
|
473
|
# the very least we check that basic text/unicode calls work OK.
|
|
470
|
m = DummyMagics(_ip)
|
|
474
|
m = DummyMagics(_ip)
|
|
471
|
nt.assert_equal(m.parse_options('foo', '')[1], 'foo')
|
|
475
|
nt.assert_equal(m.parse_options('foo', '')[1], 'foo')
|
|
472
|
nt.assert_equal(m.parse_options(u'foo', '')[1], u'foo')
|
|
476
|
nt.assert_equal(m.parse_options(u'foo', '')[1], u'foo')
|
|
473
|
|
|
477
|
|
|
474
|
|
|
478
|
|
|
475
|
def test_parse_options_preserve_non_option_string():
|
|
479
|
def test_parse_options_preserve_non_option_string():
|
|
476
|
"""Test to assert preservation of non-option part of magic-block, while parsing magic options."""
|
|
480
|
"""Test to assert preservation of non-option part of magic-block, while parsing magic options."""
|
|
477
|
m = DummyMagics(_ip)
|
|
481
|
m = DummyMagics(_ip)
|
|
478
|
opts, stmt = m.parse_options(
|
|
482
|
opts, stmt = m.parse_options(
|
|
479
|
" -n1 -r 13 _ = 314 + foo", "n:r:", preserve_non_opts=True
|
|
483
|
" -n1 -r 13 _ = 314 + foo", "n:r:", preserve_non_opts=True
|
|
480
|
)
|
|
484
|
)
|
|
481
|
nt.assert_equal(opts, {"n": "1", "r": "13"})
|
|
485
|
nt.assert_equal(opts, {"n": "1", "r": "13"})
|
|
482
|
nt.assert_equal(stmt, "_ = 314 + foo")
|
|
486
|
nt.assert_equal(stmt, "_ = 314 + foo")
|
|
483
|
|
|
487
|
|
|
484
|
|
|
488
|
|
|
485
|
def test_run_magic_preserve_code_block():
|
|
489
|
def test_run_magic_preserve_code_block():
|
|
486
|
"""Test to assert preservation of non-option part of magic-block, while running magic."""
|
|
490
|
"""Test to assert preservation of non-option part of magic-block, while running magic."""
|
|
487
|
_ip.user_ns["spaces"] = []
|
|
491
|
_ip.user_ns["spaces"] = []
|
|
488
|
_ip.magic("timeit -n1 -r1 spaces.append([s.count(' ') for s in ['document']])")
|
|
492
|
_ip.magic("timeit -n1 -r1 spaces.append([s.count(' ') for s in ['document']])")
|
|
489
|
assert _ip.user_ns["spaces"] == [[0]]
|
|
493
|
assert _ip.user_ns["spaces"] == [[0]]
|
|
490
|
|
|
494
|
|
|
491
|
|
|
495
|
|
|
492
|
def test_dirops():
|
|
496
|
def test_dirops():
|
|
493
|
"""Test various directory handling operations."""
|
|
497
|
"""Test various directory handling operations."""
|
|
494
|
# curpath = lambda :os.path.splitdrive(os.getcwd())[1].replace('\\','/')
|
|
498
|
# curpath = lambda :os.path.splitdrive(os.getcwd())[1].replace('\\','/')
|
|
495
|
curpath = os.getcwd
|
|
499
|
curpath = os.getcwd
|
|
496
|
startdir = os.getcwd()
|
|
500
|
startdir = os.getcwd()
|
|
497
|
ipdir = os.path.realpath(_ip.ipython_dir)
|
|
501
|
ipdir = os.path.realpath(_ip.ipython_dir)
|
|
498
|
try:
|
|
502
|
try:
|
|
499
|
_ip.magic('cd "%s"' % ipdir)
|
|
503
|
_ip.magic('cd "%s"' % ipdir)
|
|
500
|
nt.assert_equal(curpath(), ipdir)
|
|
504
|
nt.assert_equal(curpath(), ipdir)
|
|
501
|
_ip.magic('cd -')
|
|
505
|
_ip.magic('cd -')
|
|
502
|
nt.assert_equal(curpath(), startdir)
|
|
506
|
nt.assert_equal(curpath(), startdir)
|
|
503
|
_ip.magic('pushd "%s"' % ipdir)
|
|
507
|
_ip.magic('pushd "%s"' % ipdir)
|
|
504
|
nt.assert_equal(curpath(), ipdir)
|
|
508
|
nt.assert_equal(curpath(), ipdir)
|
|
505
|
_ip.magic('popd')
|
|
509
|
_ip.magic('popd')
|
|
506
|
nt.assert_equal(curpath(), startdir)
|
|
510
|
nt.assert_equal(curpath(), startdir)
|
|
507
|
finally:
|
|
511
|
finally:
|
|
508
|
os.chdir(startdir)
|
|
512
|
os.chdir(startdir)
|
|
509
|
|
|
513
|
|
|
510
|
|
|
514
|
|
|
511
|
def test_cd_force_quiet():
|
|
515
|
def test_cd_force_quiet():
|
|
512
|
"""Test OSMagics.cd_force_quiet option"""
|
|
516
|
"""Test OSMagics.cd_force_quiet option"""
|
|
513
|
_ip.config.OSMagics.cd_force_quiet = True
|
|
517
|
_ip.config.OSMagics.cd_force_quiet = True
|
|
514
|
osmagics = osm.OSMagics(shell=_ip)
|
|
518
|
osmagics = osm.OSMagics(shell=_ip)
|
|
515
|
|
|
519
|
|
|
516
|
startdir = os.getcwd()
|
|
520
|
startdir = os.getcwd()
|
|
517
|
ipdir = os.path.realpath(_ip.ipython_dir)
|
|
521
|
ipdir = os.path.realpath(_ip.ipython_dir)
|
|
518
|
|
|
522
|
|
|
519
|
try:
|
|
523
|
try:
|
|
520
|
with tt.AssertNotPrints(ipdir):
|
|
524
|
with tt.AssertNotPrints(ipdir):
|
|
521
|
osmagics.cd('"%s"' % ipdir)
|
|
525
|
osmagics.cd('"%s"' % ipdir)
|
|
522
|
with tt.AssertNotPrints(startdir):
|
|
526
|
with tt.AssertNotPrints(startdir):
|
|
523
|
osmagics.cd('-')
|
|
527
|
osmagics.cd('-')
|
|
524
|
finally:
|
|
528
|
finally:
|
|
525
|
os.chdir(startdir)
|
|
529
|
os.chdir(startdir)
|
|
526
|
|
|
530
|
|
|
527
|
|
|
531
|
|
|
528
|
def test_xmode():
|
|
532
|
def test_xmode():
|
|
529
|
# Calling xmode three times should be a no-op
|
|
533
|
# Calling xmode three times should be a no-op
|
|
530
|
xmode = _ip.InteractiveTB.mode
|
|
534
|
xmode = _ip.InteractiveTB.mode
|
|
531
|
for i in range(4):
|
|
535
|
for i in range(4):
|
|
532
|
_ip.magic("xmode")
|
|
536
|
_ip.magic("xmode")
|
|
533
|
nt.assert_equal(_ip.InteractiveTB.mode, xmode)
|
|
537
|
nt.assert_equal(_ip.InteractiveTB.mode, xmode)
|
|
534
|
|
|
538
|
|
|
535
|
def test_reset_hard():
|
|
539
|
def test_reset_hard():
|
|
536
|
monitor = []
|
|
540
|
monitor = []
|
|
537
|
class A(object):
|
|
541
|
class A(object):
|
|
538
|
def __del__(self):
|
|
542
|
def __del__(self):
|
|
539
|
monitor.append(1)
|
|
543
|
monitor.append(1)
|
|
540
|
def __repr__(self):
|
|
544
|
def __repr__(self):
|
|
541
|
return "<A instance>"
|
|
545
|
return "<A instance>"
|
|
542
|
|
|
546
|
|
|
543
|
_ip.user_ns["a"] = A()
|
|
547
|
_ip.user_ns["a"] = A()
|
|
544
|
_ip.run_cell("a")
|
|
548
|
_ip.run_cell("a")
|
|
545
|
|
|
549
|
|
|
546
|
nt.assert_equal(monitor, [])
|
|
550
|
nt.assert_equal(monitor, [])
|
|
547
|
_ip.magic("reset -f")
|
|
551
|
_ip.magic("reset -f")
|
|
548
|
nt.assert_equal(monitor, [1])
|
|
552
|
nt.assert_equal(monitor, [1])
|
|
549
|
|
|
553
|
|
|
550
|
class TestXdel(tt.TempFileMixin):
|
|
554
|
class TestXdel(tt.TempFileMixin):
|
|
551
|
def test_xdel(self):
|
|
555
|
def test_xdel(self):
|
|
552
|
"""Test that references from %run are cleared by xdel."""
|
|
556
|
"""Test that references from %run are cleared by xdel."""
|
|
553
|
src = ("class A(object):\n"
|
|
557
|
src = ("class A(object):\n"
|
|
554
|
" monitor = []\n"
|
|
558
|
" monitor = []\n"
|
|
555
|
" def __del__(self):\n"
|
|
559
|
" def __del__(self):\n"
|
|
556
|
" self.monitor.append(1)\n"
|
|
560
|
" self.monitor.append(1)\n"
|
|
557
|
"a = A()\n")
|
|
561
|
"a = A()\n")
|
|
558
|
self.mktmp(src)
|
|
562
|
self.mktmp(src)
|
|
559
|
# %run creates some hidden references...
|
|
563
|
# %run creates some hidden references...
|
|
560
|
_ip.magic("run %s" % self.fname)
|
|
564
|
_ip.magic("run %s" % self.fname)
|
|
561
|
# ... as does the displayhook.
|
|
565
|
# ... as does the displayhook.
|
|
562
|
_ip.run_cell("a")
|
|
566
|
_ip.run_cell("a")
|
|
563
|
|
|
567
|
|
|
564
|
monitor = _ip.user_ns["A"].monitor
|
|
568
|
monitor = _ip.user_ns["A"].monitor
|
|
565
|
nt.assert_equal(monitor, [])
|
|
569
|
nt.assert_equal(monitor, [])
|
|
566
|
|
|
570
|
|
|
567
|
_ip.magic("xdel a")
|
|
571
|
_ip.magic("xdel a")
|
|
568
|
|
|
572
|
|
|
569
|
# Check that a's __del__ method has been called.
|
|
573
|
# Check that a's __del__ method has been called.
|
|
570
|
nt.assert_equal(monitor, [1])
|
|
574
|
nt.assert_equal(monitor, [1])
|
|
571
|
|
|
575
|
|
|
572
|
def doctest_who():
|
|
576
|
def doctest_who():
|
|
573
|
"""doctest for %who
|
|
577
|
"""doctest for %who
|
|
574
|
|
|
578
|
|
|
575
|
In [1]: %reset -f
|
|
579
|
In [1]: %reset -f
|
|
576
|
|
|
580
|
|
|
577
|
In [2]: alpha = 123
|
|
581
|
In [2]: alpha = 123
|
|
578
|
|
|
582
|
|
|
579
|
In [3]: beta = 'beta'
|
|
583
|
In [3]: beta = 'beta'
|
|
580
|
|
|
584
|
|
|
581
|
In [4]: %who int
|
|
585
|
In [4]: %who int
|
|
582
|
alpha
|
|
586
|
alpha
|
|
583
|
|
|
587
|
|
|
584
|
In [5]: %who str
|
|
588
|
In [5]: %who str
|
|
585
|
beta
|
|
589
|
beta
|
|
586
|
|
|
590
|
|
|
587
|
In [6]: %whos
|
|
591
|
In [6]: %whos
|
|
588
|
Variable Type Data/Info
|
|
592
|
Variable Type Data/Info
|
|
589
|
----------------------------
|
|
593
|
----------------------------
|
|
590
|
alpha int 123
|
|
594
|
alpha int 123
|
|
591
|
beta str beta
|
|
595
|
beta str beta
|
|
592
|
|
|
596
|
|
|
593
|
In [7]: %who_ls
|
|
597
|
In [7]: %who_ls
|
|
594
|
Out[7]: ['alpha', 'beta']
|
|
598
|
Out[7]: ['alpha', 'beta']
|
|
595
|
"""
|
|
599
|
"""
|
|
596
|
|
|
600
|
|
|
597
|
def test_whos():
|
|
601
|
def test_whos():
|
|
598
|
"""Check that whos is protected against objects where repr() fails."""
|
|
602
|
"""Check that whos is protected against objects where repr() fails."""
|
|
599
|
class A(object):
|
|
603
|
class A(object):
|
|
600
|
def __repr__(self):
|
|
604
|
def __repr__(self):
|
|
601
|
raise Exception()
|
|
605
|
raise Exception()
|
|
602
|
_ip.user_ns['a'] = A()
|
|
606
|
_ip.user_ns['a'] = A()
|
|
603
|
_ip.magic("whos")
|
|
607
|
_ip.magic("whos")
|
|
604
|
|
|
608
|
|
|
605
|
def doctest_precision():
|
|
609
|
def doctest_precision():
|
|
606
|
"""doctest for %precision
|
|
610
|
"""doctest for %precision
|
|
607
|
|
|
611
|
|
|
608
|
In [1]: f = get_ipython().display_formatter.formatters['text/plain']
|
|
612
|
In [1]: f = get_ipython().display_formatter.formatters['text/plain']
|
|
609
|
|
|
613
|
|
|
610
|
In [2]: %precision 5
|
|
614
|
In [2]: %precision 5
|
|
611
|
Out[2]: '%.5f'
|
|
615
|
Out[2]: '%.5f'
|
|
612
|
|
|
616
|
|
|
613
|
In [3]: f.float_format
|
|
617
|
In [3]: f.float_format
|
|
614
|
Out[3]: '%.5f'
|
|
618
|
Out[3]: '%.5f'
|
|
615
|
|
|
619
|
|
|
616
|
In [4]: %precision %e
|
|
620
|
In [4]: %precision %e
|
|
617
|
Out[4]: '%e'
|
|
621
|
Out[4]: '%e'
|
|
618
|
|
|
622
|
|
|
619
|
In [5]: f(3.1415927)
|
|
623
|
In [5]: f(3.1415927)
|
|
620
|
Out[5]: '3.141593e+00'
|
|
624
|
Out[5]: '3.141593e+00'
|
|
621
|
"""
|
|
625
|
"""
|
|
622
|
|
|
626
|
|
|
623
|
def test_debug_magic():
|
|
627
|
def test_debug_magic():
|
|
624
|
"""Test debugging a small code with %debug
|
|
628
|
"""Test debugging a small code with %debug
|
|
625
|
|
|
629
|
|
|
626
|
In [1]: with PdbTestInput(['c']):
|
|
630
|
In [1]: with PdbTestInput(['c']):
|
|
627
|
...: %debug print("a b") #doctest: +ELLIPSIS
|
|
631
|
...: %debug print("a b") #doctest: +ELLIPSIS
|
|
628
|
...:
|
|
632
|
...:
|
|
629
|
...
|
|
633
|
...
|
|
630
|
ipdb> c
|
|
634
|
ipdb> c
|
|
631
|
a b
|
|
635
|
a b
|
|
632
|
In [2]:
|
|
636
|
In [2]:
|
|
633
|
"""
|
|
637
|
"""
|
|
634
|
|
|
638
|
|
|
635
|
def test_psearch():
|
|
639
|
def test_psearch():
|
|
636
|
with tt.AssertPrints("dict.fromkeys"):
|
|
640
|
with tt.AssertPrints("dict.fromkeys"):
|
|
637
|
_ip.run_cell("dict.fr*?")
|
|
641
|
_ip.run_cell("dict.fr*?")
|
|
638
|
with tt.AssertPrints("Ο.is_integer"):
|
|
642
|
with tt.AssertPrints("Ο.is_integer"):
|
|
639
|
_ip.run_cell("Ο = 3.14;\nΟ.is_integ*?")
|
|
643
|
_ip.run_cell("Ο = 3.14;\nΟ.is_integ*?")
|
|
640
|
|
|
644
|
|
|
641
|
def test_timeit_shlex():
|
|
645
|
def test_timeit_shlex():
|
|
642
|
"""test shlex issues with timeit (#1109)"""
|
|
646
|
"""test shlex issues with timeit (#1109)"""
|
|
643
|
_ip.ex("def f(*a,**kw): pass")
|
|
647
|
_ip.ex("def f(*a,**kw): pass")
|
|
644
|
_ip.magic('timeit -n1 "this is a bug".count(" ")')
|
|
648
|
_ip.magic('timeit -n1 "this is a bug".count(" ")')
|
|
645
|
_ip.magic('timeit -r1 -n1 f(" ", 1)')
|
|
649
|
_ip.magic('timeit -r1 -n1 f(" ", 1)')
|
|
646
|
_ip.magic('timeit -r1 -n1 f(" ", 1, " ", 2, " ")')
|
|
650
|
_ip.magic('timeit -r1 -n1 f(" ", 1, " ", 2, " ")')
|
|
647
|
_ip.magic('timeit -r1 -n1 ("a " + "b")')
|
|
651
|
_ip.magic('timeit -r1 -n1 ("a " + "b")')
|
|
648
|
_ip.magic('timeit -r1 -n1 f("a " + "b")')
|
|
652
|
_ip.magic('timeit -r1 -n1 f("a " + "b")')
|
|
649
|
_ip.magic('timeit -r1 -n1 f("a " + "b ")')
|
|
653
|
_ip.magic('timeit -r1 -n1 f("a " + "b ")')
|
|
650
|
|
|
654
|
|
|
651
|
|
|
655
|
|
|
652
|
def test_timeit_special_syntax():
|
|
656
|
def test_timeit_special_syntax():
|
|
653
|
"Test %%timeit with IPython special syntax"
|
|
657
|
"Test %%timeit with IPython special syntax"
|
|
654
|
@register_line_magic
|
|
658
|
@register_line_magic
|
|
655
|
def lmagic(line):
|
|
659
|
def lmagic(line):
|
|
656
|
ip = get_ipython()
|
|
660
|
ip = get_ipython()
|
|
657
|
ip.user_ns['lmagic_out'] = line
|
|
661
|
ip.user_ns['lmagic_out'] = line
|
|
658
|
|
|
662
|
|
|
659
|
# line mode test
|
|
663
|
# line mode test
|
|
660
|
_ip.run_line_magic('timeit', '-n1 -r1 %lmagic my line')
|
|
664
|
_ip.run_line_magic('timeit', '-n1 -r1 %lmagic my line')
|
|
661
|
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line')
|
|
665
|
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line')
|
|
662
|
# cell mode test
|
|
666
|
# cell mode test
|
|
663
|
_ip.run_cell_magic('timeit', '-n1 -r1', '%lmagic my line2')
|
|
667
|
_ip.run_cell_magic('timeit', '-n1 -r1', '%lmagic my line2')
|
|
664
|
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line2')
|
|
668
|
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line2')
|
|
665
|
|
|
669
|
|
|
666
|
def test_timeit_return():
|
|
670
|
def test_timeit_return():
|
|
667
|
"""
|
|
671
|
"""
|
|
668
|
test whether timeit -o return object
|
|
672
|
test whether timeit -o return object
|
|
669
|
"""
|
|
673
|
"""
|
|
670
|
|
|
674
|
|
|
671
|
res = _ip.run_line_magic('timeit','-n10 -r10 -o 1')
|
|
675
|
res = _ip.run_line_magic('timeit','-n10 -r10 -o 1')
|
|
672
|
assert(res is not None)
|
|
676
|
assert(res is not None)
|
|
673
|
|
|
677
|
|
|
674
|
def test_timeit_quiet():
|
|
678
|
def test_timeit_quiet():
|
|
675
|
"""
|
|
679
|
"""
|
|
676
|
test quiet option of timeit magic
|
|
680
|
test quiet option of timeit magic
|
|
677
|
"""
|
|
681
|
"""
|
|
678
|
with tt.AssertNotPrints("loops"):
|
|
682
|
with tt.AssertNotPrints("loops"):
|
|
679
|
_ip.run_cell("%timeit -n1 -r1 -q 1")
|
|
683
|
_ip.run_cell("%timeit -n1 -r1 -q 1")
|
|
680
|
|
|
684
|
|
|
681
|
def test_timeit_return_quiet():
|
|
685
|
def test_timeit_return_quiet():
|
|
682
|
with tt.AssertNotPrints("loops"):
|
|
686
|
with tt.AssertNotPrints("loops"):
|
|
683
|
res = _ip.run_line_magic('timeit', '-n1 -r1 -q -o 1')
|
|
687
|
res = _ip.run_line_magic('timeit', '-n1 -r1 -q -o 1')
|
|
684
|
assert (res is not None)
|
|
688
|
assert (res is not None)
|
|
685
|
|
|
689
|
|
|
686
|
def test_timeit_invalid_return():
|
|
690
|
def test_timeit_invalid_return():
|
|
687
|
with nt.assert_raises_regex(SyntaxError, "outside function"):
|
|
691
|
with nt.assert_raises_regex(SyntaxError, "outside function"):
|
|
688
|
_ip.run_line_magic('timeit', 'return')
|
|
692
|
_ip.run_line_magic('timeit', 'return')
|
|
689
|
|
|
693
|
|
|
690
|
@dec.skipif(execution.profile is None)
|
|
694
|
@dec.skipif(execution.profile is None)
|
|
691
|
def test_prun_special_syntax():
|
|
695
|
def test_prun_special_syntax():
|
|
692
|
"Test %%prun with IPython special syntax"
|
|
696
|
"Test %%prun with IPython special syntax"
|
|
693
|
@register_line_magic
|
|
697
|
@register_line_magic
|
|
694
|
def lmagic(line):
|
|
698
|
def lmagic(line):
|
|
695
|
ip = get_ipython()
|
|
699
|
ip = get_ipython()
|
|
696
|
ip.user_ns['lmagic_out'] = line
|
|
700
|
ip.user_ns['lmagic_out'] = line
|
|
697
|
|
|
701
|
|
|
698
|
# line mode test
|
|
702
|
# line mode test
|
|
699
|
_ip.run_line_magic('prun', '-q %lmagic my line')
|
|
703
|
_ip.run_line_magic('prun', '-q %lmagic my line')
|
|
700
|
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line')
|
|
704
|
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line')
|
|
701
|
# cell mode test
|
|
705
|
# cell mode test
|
|
702
|
_ip.run_cell_magic('prun', '-q', '%lmagic my line2')
|
|
706
|
_ip.run_cell_magic('prun', '-q', '%lmagic my line2')
|
|
703
|
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line2')
|
|
707
|
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line2')
|
|
704
|
|
|
708
|
|
|
705
|
@dec.skipif(execution.profile is None)
|
|
709
|
@dec.skipif(execution.profile is None)
|
|
706
|
def test_prun_quotes():
|
|
710
|
def test_prun_quotes():
|
|
707
|
"Test that prun does not clobber string escapes (GH #1302)"
|
|
711
|
"Test that prun does not clobber string escapes (GH #1302)"
|
|
708
|
_ip.magic(r"prun -q x = '\t'")
|
|
712
|
_ip.magic(r"prun -q x = '\t'")
|
|
709
|
nt.assert_equal(_ip.user_ns['x'], '\t')
|
|
713
|
nt.assert_equal(_ip.user_ns['x'], '\t')
|
|
710
|
|
|
714
|
|
|
711
|
def test_extension():
|
|
715
|
def test_extension():
|
|
712
|
# Debugging information for failures of this test
|
|
716
|
# Debugging information for failures of this test
|
|
713
|
print('sys.path:')
|
|
717
|
print('sys.path:')
|
|
714
|
for p in sys.path:
|
|
718
|
for p in sys.path:
|
|
715
|
print(' ', p)
|
|
719
|
print(' ', p)
|
|
716
|
print('CWD', os.getcwd())
|
|
720
|
print('CWD', os.getcwd())
|
|
717
|
|
|
721
|
|
|
718
|
nt.assert_raises(ImportError, _ip.magic, "load_ext daft_extension")
|
|
722
|
nt.assert_raises(ImportError, _ip.magic, "load_ext daft_extension")
|
|
719
|
daft_path = os.path.join(os.path.dirname(__file__), "daft_extension")
|
|
723
|
daft_path = os.path.join(os.path.dirname(__file__), "daft_extension")
|
|
720
|
sys.path.insert(0, daft_path)
|
|
724
|
sys.path.insert(0, daft_path)
|
|
721
|
try:
|
|
725
|
try:
|
|
722
|
_ip.user_ns.pop('arq', None)
|
|
726
|
_ip.user_ns.pop('arq', None)
|
|
723
|
invalidate_caches() # Clear import caches
|
|
727
|
invalidate_caches() # Clear import caches
|
|
724
|
_ip.magic("load_ext daft_extension")
|
|
728
|
_ip.magic("load_ext daft_extension")
|
|
725
|
nt.assert_equal(_ip.user_ns['arq'], 185)
|
|
729
|
nt.assert_equal(_ip.user_ns['arq'], 185)
|
|
726
|
_ip.magic("unload_ext daft_extension")
|
|
730
|
_ip.magic("unload_ext daft_extension")
|
|
727
|
assert 'arq' not in _ip.user_ns
|
|
731
|
assert 'arq' not in _ip.user_ns
|
|
728
|
finally:
|
|
732
|
finally:
|
|
729
|
sys.path.remove(daft_path)
|
|
733
|
sys.path.remove(daft_path)
|
|
730
|
|
|
734
|
|
|
731
|
|
|
735
|
|
|
732
|
def test_notebook_export_json():
|
|
736
|
def test_notebook_export_json():
|
|
733
|
_ip = get_ipython()
|
|
737
|
_ip = get_ipython()
|
|
734
|
_ip.history_manager.reset() # Clear any existing history.
|
|
738
|
_ip.history_manager.reset() # Clear any existing history.
|
|
735
|
cmds = [u"a=1", u"def b():\n return a**2", u"print('noΓ«l, Γ©tΓ©', b())"]
|
|
739
|
cmds = [u"a=1", u"def b():\n return a**2", u"print('noΓ«l, Γ©tΓ©', b())"]
|
|
736
|
for i, cmd in enumerate(cmds, start=1):
|
|
740
|
for i, cmd in enumerate(cmds, start=1):
|
|
737
|
_ip.history_manager.store_inputs(i, cmd)
|
|
741
|
_ip.history_manager.store_inputs(i, cmd)
|
|
738
|
with TemporaryDirectory() as td:
|
|
742
|
with TemporaryDirectory() as td:
|
|
739
|
outfile = os.path.join(td, "nb.ipynb")
|
|
743
|
outfile = os.path.join(td, "nb.ipynb")
|
|
740
|
_ip.magic("notebook -e %s" % outfile)
|
|
744
|
_ip.magic("notebook -e %s" % outfile)
|
|
741
|
|
|
745
|
|
|
742
|
|
|
746
|
|
|
743
|
class TestEnv(TestCase):
|
|
747
|
class TestEnv(TestCase):
|
|
744
|
|
|
748
|
|
|
745
|
def test_env(self):
|
|
749
|
def test_env(self):
|
|
746
|
env = _ip.magic("env")
|
|
750
|
env = _ip.magic("env")
|
|
747
|
self.assertTrue(isinstance(env, dict))
|
|
751
|
self.assertTrue(isinstance(env, dict))
|
|
748
|
|
|
752
|
|
|
749
|
def test_env_secret(self):
|
|
753
|
def test_env_secret(self):
|
|
750
|
env = _ip.magic("env")
|
|
754
|
env = _ip.magic("env")
|
|
751
|
hidden = "<hidden>"
|
|
755
|
hidden = "<hidden>"
|
|
752
|
with mock.patch.dict(
|
|
756
|
with mock.patch.dict(
|
|
753
|
os.environ,
|
|
757
|
os.environ,
|
|
754
|
{
|
|
758
|
{
|
|
755
|
"API_KEY": "abc123",
|
|
759
|
"API_KEY": "abc123",
|
|
756
|
"SECRET_THING": "ssshhh",
|
|
760
|
"SECRET_THING": "ssshhh",
|
|
757
|
"JUPYTER_TOKEN": "",
|
|
761
|
"JUPYTER_TOKEN": "",
|
|
758
|
"VAR": "abc"
|
|
762
|
"VAR": "abc"
|
|
759
|
}
|
|
763
|
}
|
|
760
|
):
|
|
764
|
):
|
|
761
|
env = _ip.magic("env")
|
|
765
|
env = _ip.magic("env")
|
|
762
|
assert env["API_KEY"] == hidden
|
|
766
|
assert env["API_KEY"] == hidden
|
|
763
|
assert env["SECRET_THING"] == hidden
|
|
767
|
assert env["SECRET_THING"] == hidden
|
|
764
|
assert env["JUPYTER_TOKEN"] == hidden
|
|
768
|
assert env["JUPYTER_TOKEN"] == hidden
|
|
765
|
assert env["VAR"] == "abc"
|
|
769
|
assert env["VAR"] == "abc"
|
|
766
|
|
|
770
|
|
|
767
|
def test_env_get_set_simple(self):
|
|
771
|
def test_env_get_set_simple(self):
|
|
768
|
env = _ip.magic("env var val1")
|
|
772
|
env = _ip.magic("env var val1")
|
|
769
|
self.assertEqual(env, None)
|
|
773
|
self.assertEqual(env, None)
|
|
770
|
self.assertEqual(os.environ['var'], 'val1')
|
|
774
|
self.assertEqual(os.environ['var'], 'val1')
|
|
771
|
self.assertEqual(_ip.magic("env var"), 'val1')
|
|
775
|
self.assertEqual(_ip.magic("env var"), 'val1')
|
|
772
|
env = _ip.magic("env var=val2")
|
|
776
|
env = _ip.magic("env var=val2")
|
|
773
|
self.assertEqual(env, None)
|
|
777
|
self.assertEqual(env, None)
|
|
774
|
self.assertEqual(os.environ['var'], 'val2')
|
|
778
|
self.assertEqual(os.environ['var'], 'val2')
|
|
775
|
|
|
779
|
|
|
776
|
def test_env_get_set_complex(self):
|
|
780
|
def test_env_get_set_complex(self):
|
|
777
|
env = _ip.magic("env var 'val1 '' 'val2")
|
|
781
|
env = _ip.magic("env var 'val1 '' 'val2")
|
|
778
|
self.assertEqual(env, None)
|
|
782
|
self.assertEqual(env, None)
|
|
779
|
self.assertEqual(os.environ['var'], "'val1 '' 'val2")
|
|
783
|
self.assertEqual(os.environ['var'], "'val1 '' 'val2")
|
|
780
|
self.assertEqual(_ip.magic("env var"), "'val1 '' 'val2")
|
|
784
|
self.assertEqual(_ip.magic("env var"), "'val1 '' 'val2")
|
|
781
|
env = _ip.magic('env var=val2 val3="val4')
|
|
785
|
env = _ip.magic('env var=val2 val3="val4')
|
|
782
|
self.assertEqual(env, None)
|
|
786
|
self.assertEqual(env, None)
|
|
783
|
self.assertEqual(os.environ['var'], 'val2 val3="val4')
|
|
787
|
self.assertEqual(os.environ['var'], 'val2 val3="val4')
|
|
784
|
|
|
788
|
|
|
785
|
def test_env_set_bad_input(self):
|
|
789
|
def test_env_set_bad_input(self):
|
|
786
|
self.assertRaises(UsageError, lambda: _ip.magic("set_env var"))
|
|
790
|
self.assertRaises(UsageError, lambda: _ip.magic("set_env var"))
|
|
787
|
|
|
791
|
|
|
788
|
def test_env_set_whitespace(self):
|
|
792
|
def test_env_set_whitespace(self):
|
|
789
|
self.assertRaises(UsageError, lambda: _ip.magic("env var A=B"))
|
|
793
|
self.assertRaises(UsageError, lambda: _ip.magic("env var A=B"))
|
|
790
|
|
|
794
|
|
|
791
|
|
|
795
|
|
|
792
|
class CellMagicTestCase(TestCase):
|
|
796
|
class CellMagicTestCase(TestCase):
|
|
793
|
|
|
797
|
|
|
794
|
def check_ident(self, magic):
|
|
798
|
def check_ident(self, magic):
|
|
795
|
# Manually called, we get the result
|
|
799
|
# Manually called, we get the result
|
|
796
|
out = _ip.run_cell_magic(magic, 'a', 'b')
|
|
800
|
out = _ip.run_cell_magic(magic, 'a', 'b')
|
|
797
|
nt.assert_equal(out, ('a','b'))
|
|
801
|
nt.assert_equal(out, ('a','b'))
|
|
798
|
# Via run_cell, it goes into the user's namespace via displayhook
|
|
802
|
# Via run_cell, it goes into the user's namespace via displayhook
|
|
799
|
_ip.run_cell('%%' + magic +' c\nd\n')
|
|
803
|
_ip.run_cell('%%' + magic +' c\nd\n')
|
|
800
|
nt.assert_equal(_ip.user_ns['_'], ('c','d\n'))
|
|
804
|
nt.assert_equal(_ip.user_ns['_'], ('c','d\n'))
|
|
801
|
|
|
805
|
|
|
802
|
def test_cell_magic_func_deco(self):
|
|
806
|
def test_cell_magic_func_deco(self):
|
|
803
|
"Cell magic using simple decorator"
|
|
807
|
"Cell magic using simple decorator"
|
|
804
|
@register_cell_magic
|
|
808
|
@register_cell_magic
|
|
805
|
def cellm(line, cell):
|
|
809
|
def cellm(line, cell):
|
|
806
|
return line, cell
|
|
810
|
return line, cell
|
|
807
|
|
|
811
|
|
|
808
|
self.check_ident('cellm')
|
|
812
|
self.check_ident('cellm')
|
|
809
|
|
|
813
|
|
|
810
|
def test_cell_magic_reg(self):
|
|
814
|
def test_cell_magic_reg(self):
|
|
811
|
"Cell magic manually registered"
|
|
815
|
"Cell magic manually registered"
|
|
812
|
def cellm(line, cell):
|
|
816
|
def cellm(line, cell):
|
|
813
|
return line, cell
|
|
817
|
return line, cell
|
|
814
|
|
|
818
|
|
|
815
|
_ip.register_magic_function(cellm, 'cell', 'cellm2')
|
|
819
|
_ip.register_magic_function(cellm, 'cell', 'cellm2')
|
|
816
|
self.check_ident('cellm2')
|
|
820
|
self.check_ident('cellm2')
|
|
817
|
|
|
821
|
|
|
818
|
def test_cell_magic_class(self):
|
|
822
|
def test_cell_magic_class(self):
|
|
819
|
"Cell magics declared via a class"
|
|
823
|
"Cell magics declared via a class"
|
|
820
|
@magics_class
|
|
824
|
@magics_class
|
|
821
|
class MyMagics(Magics):
|
|
825
|
class MyMagics(Magics):
|
|
822
|
|
|
826
|
|
|
823
|
@cell_magic
|
|
827
|
@cell_magic
|
|
824
|
def cellm3(self, line, cell):
|
|
828
|
def cellm3(self, line, cell):
|
|
825
|
return line, cell
|
|
829
|
return line, cell
|
|
826
|
|
|
830
|
|
|
827
|
_ip.register_magics(MyMagics)
|
|
831
|
_ip.register_magics(MyMagics)
|
|
828
|
self.check_ident('cellm3')
|
|
832
|
self.check_ident('cellm3')
|
|
829
|
|
|
833
|
|
|
830
|
def test_cell_magic_class2(self):
|
|
834
|
def test_cell_magic_class2(self):
|
|
831
|
"Cell magics declared via a class, #2"
|
|
835
|
"Cell magics declared via a class, #2"
|
|
832
|
@magics_class
|
|
836
|
@magics_class
|
|
833
|
class MyMagics2(Magics):
|
|
837
|
class MyMagics2(Magics):
|
|
834
|
|
|
838
|
|
|
835
|
@cell_magic('cellm4')
|
|
839
|
@cell_magic('cellm4')
|
|
836
|
def cellm33(self, line, cell):
|
|
840
|
def cellm33(self, line, cell):
|
|
837
|
return line, cell
|
|
841
|
return line, cell
|
|
838
|
|
|
842
|
|
|
839
|
_ip.register_magics(MyMagics2)
|
|
843
|
_ip.register_magics(MyMagics2)
|
|
840
|
self.check_ident('cellm4')
|
|
844
|
self.check_ident('cellm4')
|
|
841
|
# Check that nothing is registered as 'cellm33'
|
|
845
|
# Check that nothing is registered as 'cellm33'
|
|
842
|
c33 = _ip.find_cell_magic('cellm33')
|
|
846
|
c33 = _ip.find_cell_magic('cellm33')
|
|
843
|
nt.assert_equal(c33, None)
|
|
847
|
nt.assert_equal(c33, None)
|
|
844
|
|
|
848
|
|
|
845
|
def test_file():
|
|
849
|
def test_file():
|
|
846
|
"""Basic %%writefile"""
|
|
850
|
"""Basic %%writefile"""
|
|
847
|
ip = get_ipython()
|
|
851
|
ip = get_ipython()
|
|
848
|
with TemporaryDirectory() as td:
|
|
852
|
with TemporaryDirectory() as td:
|
|
849
|
fname = os.path.join(td, 'file1')
|
|
853
|
fname = os.path.join(td, 'file1')
|
|
850
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
854
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
851
|
'line1',
|
|
855
|
'line1',
|
|
852
|
'line2',
|
|
856
|
'line2',
|
|
853
|
]))
|
|
857
|
]))
|
|
854
|
s = Path(fname).read_text()
|
|
858
|
s = Path(fname).read_text()
|
|
855
|
nt.assert_in('line1\n', s)
|
|
859
|
nt.assert_in('line1\n', s)
|
|
856
|
nt.assert_in('line2', s)
|
|
860
|
nt.assert_in('line2', s)
|
|
857
|
|
|
861
|
|
|
858
|
@dec.skip_win32
|
|
862
|
@dec.skip_win32
|
|
859
|
def test_file_single_quote():
|
|
863
|
def test_file_single_quote():
|
|
860
|
"""Basic %%writefile with embedded single quotes"""
|
|
864
|
"""Basic %%writefile with embedded single quotes"""
|
|
861
|
ip = get_ipython()
|
|
865
|
ip = get_ipython()
|
|
862
|
with TemporaryDirectory() as td:
|
|
866
|
with TemporaryDirectory() as td:
|
|
863
|
fname = os.path.join(td, '\'file1\'')
|
|
867
|
fname = os.path.join(td, '\'file1\'')
|
|
864
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
868
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
865
|
'line1',
|
|
869
|
'line1',
|
|
866
|
'line2',
|
|
870
|
'line2',
|
|
867
|
]))
|
|
871
|
]))
|
|
868
|
s = Path(fname).read_text()
|
|
872
|
s = Path(fname).read_text()
|
|
869
|
nt.assert_in('line1\n', s)
|
|
873
|
nt.assert_in('line1\n', s)
|
|
870
|
nt.assert_in('line2', s)
|
|
874
|
nt.assert_in('line2', s)
|
|
871
|
|
|
875
|
|
|
872
|
@dec.skip_win32
|
|
876
|
@dec.skip_win32
|
|
873
|
def test_file_double_quote():
|
|
877
|
def test_file_double_quote():
|
|
874
|
"""Basic %%writefile with embedded double quotes"""
|
|
878
|
"""Basic %%writefile with embedded double quotes"""
|
|
875
|
ip = get_ipython()
|
|
879
|
ip = get_ipython()
|
|
876
|
with TemporaryDirectory() as td:
|
|
880
|
with TemporaryDirectory() as td:
|
|
877
|
fname = os.path.join(td, '"file1"')
|
|
881
|
fname = os.path.join(td, '"file1"')
|
|
878
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
882
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
879
|
'line1',
|
|
883
|
'line1',
|
|
880
|
'line2',
|
|
884
|
'line2',
|
|
881
|
]))
|
|
885
|
]))
|
|
882
|
s = Path(fname).read_text()
|
|
886
|
s = Path(fname).read_text()
|
|
883
|
nt.assert_in('line1\n', s)
|
|
887
|
nt.assert_in('line1\n', s)
|
|
884
|
nt.assert_in('line2', s)
|
|
888
|
nt.assert_in('line2', s)
|
|
885
|
|
|
889
|
|
|
886
|
def test_file_var_expand():
|
|
890
|
def test_file_var_expand():
|
|
887
|
"""%%writefile $filename"""
|
|
891
|
"""%%writefile $filename"""
|
|
888
|
ip = get_ipython()
|
|
892
|
ip = get_ipython()
|
|
889
|
with TemporaryDirectory() as td:
|
|
893
|
with TemporaryDirectory() as td:
|
|
890
|
fname = os.path.join(td, 'file1')
|
|
894
|
fname = os.path.join(td, 'file1')
|
|
891
|
ip.user_ns['filename'] = fname
|
|
895
|
ip.user_ns['filename'] = fname
|
|
892
|
ip.run_cell_magic("writefile", '$filename', u'\n'.join([
|
|
896
|
ip.run_cell_magic("writefile", '$filename', u'\n'.join([
|
|
893
|
'line1',
|
|
897
|
'line1',
|
|
894
|
'line2',
|
|
898
|
'line2',
|
|
895
|
]))
|
|
899
|
]))
|
|
896
|
s = Path(fname).read_text()
|
|
900
|
s = Path(fname).read_text()
|
|
897
|
nt.assert_in('line1\n', s)
|
|
901
|
nt.assert_in('line1\n', s)
|
|
898
|
nt.assert_in('line2', s)
|
|
902
|
nt.assert_in('line2', s)
|
|
899
|
|
|
903
|
|
|
900
|
def test_file_unicode():
|
|
904
|
def test_file_unicode():
|
|
901
|
"""%%writefile with unicode cell"""
|
|
905
|
"""%%writefile with unicode cell"""
|
|
902
|
ip = get_ipython()
|
|
906
|
ip = get_ipython()
|
|
903
|
with TemporaryDirectory() as td:
|
|
907
|
with TemporaryDirectory() as td:
|
|
904
|
fname = os.path.join(td, 'file1')
|
|
908
|
fname = os.path.join(td, 'file1')
|
|
905
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
909
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
906
|
u'linΓ©1',
|
|
910
|
u'linΓ©1',
|
|
907
|
u'linΓ©2',
|
|
911
|
u'linΓ©2',
|
|
908
|
]))
|
|
912
|
]))
|
|
909
|
with io.open(fname, encoding='utf-8') as f:
|
|
913
|
with io.open(fname, encoding='utf-8') as f:
|
|
910
|
s = f.read()
|
|
914
|
s = f.read()
|
|
911
|
nt.assert_in(u'linΓ©1\n', s)
|
|
915
|
nt.assert_in(u'linΓ©1\n', s)
|
|
912
|
nt.assert_in(u'linΓ©2', s)
|
|
916
|
nt.assert_in(u'linΓ©2', s)
|
|
913
|
|
|
917
|
|
|
914
|
def test_file_amend():
|
|
918
|
def test_file_amend():
|
|
915
|
"""%%writefile -a amends files"""
|
|
919
|
"""%%writefile -a amends files"""
|
|
916
|
ip = get_ipython()
|
|
920
|
ip = get_ipython()
|
|
917
|
with TemporaryDirectory() as td:
|
|
921
|
with TemporaryDirectory() as td:
|
|
918
|
fname = os.path.join(td, 'file2')
|
|
922
|
fname = os.path.join(td, 'file2')
|
|
919
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
923
|
ip.run_cell_magic("writefile", fname, u'\n'.join([
|
|
920
|
'line1',
|
|
924
|
'line1',
|
|
921
|
'line2',
|
|
925
|
'line2',
|
|
922
|
]))
|
|
926
|
]))
|
|
923
|
ip.run_cell_magic("writefile", "-a %s" % fname, u'\n'.join([
|
|
927
|
ip.run_cell_magic("writefile", "-a %s" % fname, u'\n'.join([
|
|
924
|
'line3',
|
|
928
|
'line3',
|
|
925
|
'line4',
|
|
929
|
'line4',
|
|
926
|
]))
|
|
930
|
]))
|
|
927
|
s = Path(fname).read_text()
|
|
931
|
s = Path(fname).read_text()
|
|
928
|
nt.assert_in('line1\n', s)
|
|
932
|
nt.assert_in('line1\n', s)
|
|
929
|
nt.assert_in('line3\n', s)
|
|
933
|
nt.assert_in('line3\n', s)
|
|
930
|
|
|
934
|
|
|
931
|
def test_file_spaces():
|
|
935
|
def test_file_spaces():
|
|
932
|
"""%%file with spaces in filename"""
|
|
936
|
"""%%file with spaces in filename"""
|
|
933
|
ip = get_ipython()
|
|
937
|
ip = get_ipython()
|
|
934
|
with TemporaryWorkingDirectory() as td:
|
|
938
|
with TemporaryWorkingDirectory() as td:
|
|
935
|
fname = "file name"
|
|
939
|
fname = "file name"
|
|
936
|
ip.run_cell_magic("file", '"%s"'%fname, u'\n'.join([
|
|
940
|
ip.run_cell_magic("file", '"%s"'%fname, u'\n'.join([
|
|
937
|
'line1',
|
|
941
|
'line1',
|
|
938
|
'line2',
|
|
942
|
'line2',
|
|
939
|
]))
|
|
943
|
]))
|
|
940
|
s = Path(fname).read_text()
|
|
944
|
s = Path(fname).read_text()
|
|
941
|
nt.assert_in('line1\n', s)
|
|
945
|
nt.assert_in('line1\n', s)
|
|
942
|
nt.assert_in('line2', s)
|
|
946
|
nt.assert_in('line2', s)
|
|
943
|
|
|
947
|
|
|
944
|
def test_script_config():
|
|
948
|
def test_script_config():
|
|
945
|
ip = get_ipython()
|
|
949
|
ip = get_ipython()
|
|
946
|
ip.config.ScriptMagics.script_magics = ['whoda']
|
|
950
|
ip.config.ScriptMagics.script_magics = ['whoda']
|
|
947
|
sm = script.ScriptMagics(shell=ip)
|
|
951
|
sm = script.ScriptMagics(shell=ip)
|
|
948
|
nt.assert_in('whoda', sm.magics['cell'])
|
|
952
|
nt.assert_in('whoda', sm.magics['cell'])
|
|
949
|
|
|
953
|
|
|
|
|
|
954
|
@dec.skip_iptest_but_not_pytest
|
|
950
|
@dec.skip_win32
|
|
955
|
@dec.skip_win32
|
|
|
|
|
956
|
@pytest.mark.skipif(
|
|
|
|
|
957
|
sys.platform == "win32", reason="This test does not run under Windows"
|
|
|
|
|
958
|
)
|
|
951
|
def test_script_out():
|
|
959
|
def test_script_out():
|
|
|
|
|
960
|
assert asyncio.get_event_loop().is_running() is False
|
|
|
|
|
961
|
|
|
952
|
ip = get_ipython()
|
|
962
|
ip = get_ipython()
|
|
953
|
ip.run_cell_magic("script", "--out output sh", "echo 'hi'")
|
|
963
|
ip.run_cell_magic("script", "--out output sh", "echo 'hi'")
|
|
|
|
|
964
|
assert asyncio.get_event_loop().is_running() is False
|
|
954
|
nt.assert_equal(ip.user_ns['output'], 'hi\n')
|
|
965
|
nt.assert_equal(ip.user_ns['output'], 'hi\n')
|
|
955
|
|
|
966
|
|
|
|
|
|
967
|
@dec.skip_iptest_but_not_pytest
|
|
956
|
@dec.skip_win32
|
|
968
|
@dec.skip_win32
|
|
|
|
|
969
|
@pytest.mark.skipif(
|
|
|
|
|
970
|
sys.platform == "win32", reason="This test does not run under Windows"
|
|
|
|
|
971
|
)
|
|
957
|
def test_script_err():
|
|
972
|
def test_script_err():
|
|
958
|
ip = get_ipython()
|
|
973
|
ip = get_ipython()
|
|
|
|
|
974
|
assert asyncio.get_event_loop().is_running() is False
|
|
959
|
ip.run_cell_magic("script", "--err error sh", "echo 'hello' >&2")
|
|
975
|
ip.run_cell_magic("script", "--err error sh", "echo 'hello' >&2")
|
|
|
|
|
976
|
assert asyncio.get_event_loop().is_running() is False
|
|
960
|
nt.assert_equal(ip.user_ns['error'], 'hello\n')
|
|
977
|
nt.assert_equal(ip.user_ns['error'], 'hello\n')
|
|
961
|
|
|
978
|
|
|
|
|
|
979
|
|
|
|
|
|
980
|
@dec.skip_iptest_but_not_pytest
|
|
962
|
@dec.skip_win32
|
|
981
|
@dec.skip_win32
|
|
|
|
|
982
|
@pytest.mark.skipif(
|
|
|
|
|
983
|
sys.platform == "win32", reason="This test does not run under Windows"
|
|
|
|
|
984
|
)
|
|
963
|
def test_script_out_err():
|
|
985
|
def test_script_out_err():
|
|
|
|
|
986
|
|
|
964
|
ip = get_ipython()
|
|
987
|
ip = get_ipython()
|
|
965
|
ip.run_cell_magic("script", "--out output --err error sh", "echo 'hi'\necho 'hello' >&2")
|
|
988
|
ip.run_cell_magic(
|
|
966
|
nt.assert_equal(ip.user_ns['output'], 'hi\n')
|
|
989
|
"script", "--out output --err error sh", "echo 'hi'\necho 'hello' >&2"
|
|
967
|
nt.assert_equal(ip.user_ns['error'], 'hello\n')
|
|
990
|
)
|
|
|
|
|
991
|
nt.assert_equal(ip.user_ns["output"], "hi\n")
|
|
|
|
|
992
|
nt.assert_equal(ip.user_ns["error"], "hello\n")
|
|
|
|
|
993
|
|
|
968
|
|
|
994
|
|
|
|
|
|
995
|
@dec.skip_iptest_but_not_pytest
|
|
969
|
@dec.skip_win32
|
|
996
|
@dec.skip_win32
|
|
|
|
|
997
|
@pytest.mark.skipif(
|
|
|
|
|
998
|
sys.platform == "win32", reason="This test does not run under Windows"
|
|
|
|
|
999
|
)
|
|
970
|
async def test_script_bg_out():
|
|
1000
|
async def test_script_bg_out():
|
|
971
|
ip = get_ipython()
|
|
1001
|
ip = get_ipython()
|
|
972
|
ip.run_cell_magic("script", "--bg --out output sh", "echo 'hi'")
|
|
1002
|
ip.run_cell_magic("script", "--bg --out output sh", "echo 'hi'")
|
|
973
|
nt.assert_equal((await ip.user_ns["output"].read()), b"hi\n")
|
|
1003
|
nt.assert_equal((await ip.user_ns["output"].read()), b"hi\n")
|
|
974
|
ip.user_ns['output'].close()
|
|
1004
|
ip.user_ns["output"].close()
|
|
|
|
|
1005
|
asyncio.get_event_loop().stop()
|
|
975
|
|
|
1006
|
|
|
|
|
|
1007
|
@dec.skip_iptest_but_not_pytest
|
|
976
|
@dec.skip_win32
|
|
1008
|
@dec.skip_win32
|
|
|
|
|
1009
|
@pytest.mark.skipif(
|
|
|
|
|
1010
|
sys.platform == "win32", reason="This test does not run under Windows"
|
|
|
|
|
1011
|
)
|
|
977
|
async def test_script_bg_err():
|
|
1012
|
async def test_script_bg_err():
|
|
978
|
ip = get_ipython()
|
|
1013
|
ip = get_ipython()
|
|
979
|
ip.run_cell_magic("script", "--bg --err error sh", "echo 'hello' >&2")
|
|
1014
|
ip.run_cell_magic("script", "--bg --err error sh", "echo 'hello' >&2")
|
|
980
|
nt.assert_equal((await ip.user_ns["error"].read()), b"hello\n")
|
|
1015
|
nt.assert_equal((await ip.user_ns["error"].read()), b"hello\n")
|
|
981
|
ip.user_ns["error"].close()
|
|
1016
|
ip.user_ns["error"].close()
|
|
982
|
|
|
1017
|
|
|
983
|
|
|
1018
|
|
|
|
|
|
1019
|
@dec.skip_iptest_but_not_pytest
|
|
984
|
@dec.skip_win32
|
|
1020
|
@dec.skip_win32
|
|
|
|
|
1021
|
@pytest.mark.skipif(
|
|
|
|
|
1022
|
sys.platform == "win32", reason="This test does not run under Windows"
|
|
|
|
|
1023
|
)
|
|
985
|
async def test_script_bg_out_err():
|
|
1024
|
async def test_script_bg_out_err():
|
|
986
|
ip = get_ipython()
|
|
1025
|
ip = get_ipython()
|
|
987
|
ip.run_cell_magic(
|
|
1026
|
ip.run_cell_magic(
|
|
988
|
"script", "--bg --out output --err error sh", "echo 'hi'\necho 'hello' >&2"
|
|
1027
|
"script", "--bg --out output --err error sh", "echo 'hi'\necho 'hello' >&2"
|
|
989
|
)
|
|
1028
|
)
|
|
990
|
nt.assert_equal((await ip.user_ns["output"].read()), b"hi\n")
|
|
1029
|
nt.assert_equal((await ip.user_ns["output"].read()), b"hi\n")
|
|
991
|
nt.assert_equal((await ip.user_ns["error"].read()), b"hello\n")
|
|
1030
|
nt.assert_equal((await ip.user_ns["error"].read()), b"hello\n")
|
|
992
|
ip.user_ns["output"].close()
|
|
1031
|
ip.user_ns["output"].close()
|
|
993
|
ip.user_ns["error"].close()
|
|
1032
|
ip.user_ns["error"].close()
|
|
994
|
|
|
1033
|
|
|
995
|
|
|
1034
|
|
|
996
|
def test_script_defaults():
|
|
1035
|
def test_script_defaults():
|
|
997
|
ip = get_ipython()
|
|
1036
|
ip = get_ipython()
|
|
998
|
for cmd in ['sh', 'bash', 'perl', 'ruby']:
|
|
1037
|
for cmd in ['sh', 'bash', 'perl', 'ruby']:
|
|
999
|
try:
|
|
1038
|
try:
|
|
1000
|
find_cmd(cmd)
|
|
1039
|
find_cmd(cmd)
|
|
1001
|
except Exception:
|
|
1040
|
except Exception:
|
|
1002
|
pass
|
|
1041
|
pass
|
|
1003
|
else:
|
|
1042
|
else:
|
|
1004
|
nt.assert_in(cmd, ip.magics_manager.magics['cell'])
|
|
1043
|
nt.assert_in(cmd, ip.magics_manager.magics['cell'])
|
|
1005
|
|
|
1044
|
|
|
1006
|
|
|
1045
|
|
|
1007
|
@magics_class
|
|
1046
|
@magics_class
|
|
1008
|
class FooFoo(Magics):
|
|
1047
|
class FooFoo(Magics):
|
|
1009
|
"""class with both %foo and %%foo magics"""
|
|
1048
|
"""class with both %foo and %%foo magics"""
|
|
1010
|
@line_magic('foo')
|
|
1049
|
@line_magic('foo')
|
|
1011
|
def line_foo(self, line):
|
|
1050
|
def line_foo(self, line):
|
|
1012
|
"I am line foo"
|
|
1051
|
"I am line foo"
|
|
1013
|
pass
|
|
1052
|
pass
|
|
1014
|
|
|
1053
|
|
|
1015
|
@cell_magic("foo")
|
|
1054
|
@cell_magic("foo")
|
|
1016
|
def cell_foo(self, line, cell):
|
|
1055
|
def cell_foo(self, line, cell):
|
|
1017
|
"I am cell foo, not line foo"
|
|
1056
|
"I am cell foo, not line foo"
|
|
1018
|
pass
|
|
1057
|
pass
|
|
1019
|
|
|
1058
|
|
|
1020
|
def test_line_cell_info():
|
|
1059
|
def test_line_cell_info():
|
|
1021
|
"""%%foo and %foo magics are distinguishable to inspect"""
|
|
1060
|
"""%%foo and %foo magics are distinguishable to inspect"""
|
|
1022
|
ip = get_ipython()
|
|
1061
|
ip = get_ipython()
|
|
1023
|
ip.magics_manager.register(FooFoo)
|
|
1062
|
ip.magics_manager.register(FooFoo)
|
|
1024
|
oinfo = ip.object_inspect('foo')
|
|
1063
|
oinfo = ip.object_inspect('foo')
|
|
1025
|
nt.assert_true(oinfo['found'])
|
|
1064
|
nt.assert_true(oinfo['found'])
|
|
1026
|
nt.assert_true(oinfo['ismagic'])
|
|
1065
|
nt.assert_true(oinfo['ismagic'])
|
|
1027
|
|
|
1066
|
|
|
1028
|
oinfo = ip.object_inspect('%%foo')
|
|
1067
|
oinfo = ip.object_inspect('%%foo')
|
|
1029
|
nt.assert_true(oinfo['found'])
|
|
1068
|
nt.assert_true(oinfo['found'])
|
|
1030
|
nt.assert_true(oinfo['ismagic'])
|
|
1069
|
nt.assert_true(oinfo['ismagic'])
|
|
1031
|
nt.assert_equal(oinfo['docstring'], FooFoo.cell_foo.__doc__)
|
|
1070
|
nt.assert_equal(oinfo['docstring'], FooFoo.cell_foo.__doc__)
|
|
1032
|
|
|
1071
|
|
|
1033
|
oinfo = ip.object_inspect('%foo')
|
|
1072
|
oinfo = ip.object_inspect('%foo')
|
|
1034
|
nt.assert_true(oinfo['found'])
|
|
1073
|
nt.assert_true(oinfo['found'])
|
|
1035
|
nt.assert_true(oinfo['ismagic'])
|
|
1074
|
nt.assert_true(oinfo['ismagic'])
|
|
1036
|
nt.assert_equal(oinfo['docstring'], FooFoo.line_foo.__doc__)
|
|
1075
|
nt.assert_equal(oinfo['docstring'], FooFoo.line_foo.__doc__)
|
|
1037
|
|
|
1076
|
|
|
1038
|
def test_multiple_magics():
|
|
1077
|
def test_multiple_magics():
|
|
1039
|
ip = get_ipython()
|
|
1078
|
ip = get_ipython()
|
|
1040
|
foo1 = FooFoo(ip)
|
|
1079
|
foo1 = FooFoo(ip)
|
|
1041
|
foo2 = FooFoo(ip)
|
|
1080
|
foo2 = FooFoo(ip)
|
|
1042
|
mm = ip.magics_manager
|
|
1081
|
mm = ip.magics_manager
|
|
1043
|
mm.register(foo1)
|
|
1082
|
mm.register(foo1)
|
|
1044
|
nt.assert_true(mm.magics['line']['foo'].__self__ is foo1)
|
|
1083
|
nt.assert_true(mm.magics['line']['foo'].__self__ is foo1)
|
|
1045
|
mm.register(foo2)
|
|
1084
|
mm.register(foo2)
|
|
1046
|
nt.assert_true(mm.magics['line']['foo'].__self__ is foo2)
|
|
1085
|
nt.assert_true(mm.magics['line']['foo'].__self__ is foo2)
|
|
1047
|
|
|
1086
|
|
|
1048
|
def test_alias_magic():
|
|
1087
|
def test_alias_magic():
|
|
1049
|
"""Test %alias_magic."""
|
|
1088
|
"""Test %alias_magic."""
|
|
1050
|
ip = get_ipython()
|
|
1089
|
ip = get_ipython()
|
|
1051
|
mm = ip.magics_manager
|
|
1090
|
mm = ip.magics_manager
|
|
1052
|
|
|
1091
|
|
|
1053
|
# Basic operation: both cell and line magics are created, if possible.
|
|
1092
|
# Basic operation: both cell and line magics are created, if possible.
|
|
1054
|
ip.run_line_magic('alias_magic', 'timeit_alias timeit')
|
|
1093
|
ip.run_line_magic('alias_magic', 'timeit_alias timeit')
|
|
1055
|
nt.assert_in('timeit_alias', mm.magics['line'])
|
|
1094
|
nt.assert_in('timeit_alias', mm.magics['line'])
|
|
1056
|
nt.assert_in('timeit_alias', mm.magics['cell'])
|
|
1095
|
nt.assert_in('timeit_alias', mm.magics['cell'])
|
|
1057
|
|
|
1096
|
|
|
1058
|
# --cell is specified, line magic not created.
|
|
1097
|
# --cell is specified, line magic not created.
|
|
1059
|
ip.run_line_magic('alias_magic', '--cell timeit_cell_alias timeit')
|
|
1098
|
ip.run_line_magic('alias_magic', '--cell timeit_cell_alias timeit')
|
|
1060
|
nt.assert_not_in('timeit_cell_alias', mm.magics['line'])
|
|
1099
|
nt.assert_not_in('timeit_cell_alias', mm.magics['line'])
|
|
1061
|
nt.assert_in('timeit_cell_alias', mm.magics['cell'])
|
|
1100
|
nt.assert_in('timeit_cell_alias', mm.magics['cell'])
|
|
1062
|
|
|
1101
|
|
|
1063
|
# Test that line alias is created successfully.
|
|
1102
|
# Test that line alias is created successfully.
|
|
1064
|
ip.run_line_magic('alias_magic', '--line env_alias env')
|
|
1103
|
ip.run_line_magic('alias_magic', '--line env_alias env')
|
|
1065
|
nt.assert_equal(ip.run_line_magic('env', ''),
|
|
1104
|
nt.assert_equal(ip.run_line_magic('env', ''),
|
|
1066
|
ip.run_line_magic('env_alias', ''))
|
|
1105
|
ip.run_line_magic('env_alias', ''))
|
|
1067
|
|
|
1106
|
|
|
1068
|
# Test that line alias with parameters passed in is created successfully.
|
|
1107
|
# Test that line alias with parameters passed in is created successfully.
|
|
1069
|
ip.run_line_magic('alias_magic', '--line history_alias history --params ' + shlex.quote('3'))
|
|
1108
|
ip.run_line_magic('alias_magic', '--line history_alias history --params ' + shlex.quote('3'))
|
|
1070
|
nt.assert_in('history_alias', mm.magics['line'])
|
|
1109
|
nt.assert_in('history_alias', mm.magics['line'])
|
|
1071
|
|
|
1110
|
|
|
1072
|
|
|
1111
|
|
|
1073
|
def test_save():
|
|
1112
|
def test_save():
|
|
1074
|
"""Test %save."""
|
|
1113
|
"""Test %save."""
|
|
1075
|
ip = get_ipython()
|
|
1114
|
ip = get_ipython()
|
|
1076
|
ip.history_manager.reset() # Clear any existing history.
|
|
1115
|
ip.history_manager.reset() # Clear any existing history.
|
|
1077
|
cmds = [u"a=1", u"def b():\n return a**2", u"print(a, b())"]
|
|
1116
|
cmds = [u"a=1", u"def b():\n return a**2", u"print(a, b())"]
|
|
1078
|
for i, cmd in enumerate(cmds, start=1):
|
|
1117
|
for i, cmd in enumerate(cmds, start=1):
|
|
1079
|
ip.history_manager.store_inputs(i, cmd)
|
|
1118
|
ip.history_manager.store_inputs(i, cmd)
|
|
1080
|
with TemporaryDirectory() as tmpdir:
|
|
1119
|
with TemporaryDirectory() as tmpdir:
|
|
1081
|
file = os.path.join(tmpdir, "testsave.py")
|
|
1120
|
file = os.path.join(tmpdir, "testsave.py")
|
|
1082
|
ip.run_line_magic("save", "%s 1-10" % file)
|
|
1121
|
ip.run_line_magic("save", "%s 1-10" % file)
|
|
1083
|
content = Path(file).read_text()
|
|
1122
|
content = Path(file).read_text()
|
|
1084
|
nt.assert_equal(content.count(cmds[0]), 1)
|
|
1123
|
nt.assert_equal(content.count(cmds[0]), 1)
|
|
1085
|
nt.assert_in("coding: utf-8", content)
|
|
1124
|
nt.assert_in("coding: utf-8", content)
|
|
1086
|
ip.run_line_magic("save", "-a %s 1-10" % file)
|
|
1125
|
ip.run_line_magic("save", "-a %s 1-10" % file)
|
|
1087
|
content = Path(file).read_text()
|
|
1126
|
content = Path(file).read_text()
|
|
1088
|
nt.assert_equal(content.count(cmds[0]), 2)
|
|
1127
|
nt.assert_equal(content.count(cmds[0]), 2)
|
|
1089
|
nt.assert_in("coding: utf-8", content)
|
|
1128
|
nt.assert_in("coding: utf-8", content)
|
|
1090
|
|
|
1129
|
|
|
1091
|
|
|
1130
|
|
|
1092
|
def test_save_with_no_args():
|
|
1131
|
def test_save_with_no_args():
|
|
1093
|
ip = get_ipython()
|
|
1132
|
ip = get_ipython()
|
|
1094
|
ip.history_manager.reset() # Clear any existing history.
|
|
1133
|
ip.history_manager.reset() # Clear any existing history.
|
|
1095
|
cmds = [u"a=1", u"def b():\n return a**2", u"print(a, b())", "%save"]
|
|
1134
|
cmds = [u"a=1", u"def b():\n return a**2", u"print(a, b())", "%save"]
|
|
1096
|
for i, cmd in enumerate(cmds, start=1):
|
|
1135
|
for i, cmd in enumerate(cmds, start=1):
|
|
1097
|
ip.history_manager.store_inputs(i, cmd)
|
|
1136
|
ip.history_manager.store_inputs(i, cmd)
|
|
1098
|
|
|
1137
|
|
|
1099
|
with TemporaryDirectory() as tmpdir:
|
|
1138
|
with TemporaryDirectory() as tmpdir:
|
|
1100
|
path = os.path.join(tmpdir, "testsave.py")
|
|
1139
|
path = os.path.join(tmpdir, "testsave.py")
|
|
1101
|
ip.run_line_magic("save", path)
|
|
1140
|
ip.run_line_magic("save", path)
|
|
1102
|
content = Path(path).read_text()
|
|
1141
|
content = Path(path).read_text()
|
|
1103
|
expected_content = dedent(
|
|
1142
|
expected_content = dedent(
|
|
1104
|
"""\
|
|
1143
|
"""\
|
|
1105
|
# coding: utf-8
|
|
1144
|
# coding: utf-8
|
|
1106
|
a=1
|
|
1145
|
a=1
|
|
1107
|
def b():
|
|
1146
|
def b():
|
|
1108
|
return a**2
|
|
1147
|
return a**2
|
|
1109
|
print(a, b())
|
|
1148
|
print(a, b())
|
|
1110
|
"""
|
|
1149
|
"""
|
|
1111
|
)
|
|
1150
|
)
|
|
1112
|
nt.assert_equal(content, expected_content)
|
|
1151
|
nt.assert_equal(content, expected_content)
|
|
1113
|
|
|
1152
|
|
|
1114
|
|
|
1153
|
|
|
1115
|
def test_store():
|
|
1154
|
def test_store():
|
|
1116
|
"""Test %store."""
|
|
1155
|
"""Test %store."""
|
|
1117
|
ip = get_ipython()
|
|
1156
|
ip = get_ipython()
|
|
1118
|
ip.run_line_magic('load_ext', 'storemagic')
|
|
1157
|
ip.run_line_magic('load_ext', 'storemagic')
|
|
1119
|
|
|
1158
|
|
|
1120
|
# make sure the storage is empty
|
|
1159
|
# make sure the storage is empty
|
|
1121
|
ip.run_line_magic('store', '-z')
|
|
1160
|
ip.run_line_magic('store', '-z')
|
|
1122
|
ip.user_ns['var'] = 42
|
|
1161
|
ip.user_ns['var'] = 42
|
|
1123
|
ip.run_line_magic('store', 'var')
|
|
1162
|
ip.run_line_magic('store', 'var')
|
|
1124
|
ip.user_ns['var'] = 39
|
|
1163
|
ip.user_ns['var'] = 39
|
|
1125
|
ip.run_line_magic('store', '-r')
|
|
1164
|
ip.run_line_magic('store', '-r')
|
|
1126
|
nt.assert_equal(ip.user_ns['var'], 42)
|
|
1165
|
nt.assert_equal(ip.user_ns['var'], 42)
|
|
1127
|
|
|
1166
|
|
|
1128
|
ip.run_line_magic('store', '-d var')
|
|
1167
|
ip.run_line_magic('store', '-d var')
|
|
1129
|
ip.user_ns['var'] = 39
|
|
1168
|
ip.user_ns['var'] = 39
|
|
1130
|
ip.run_line_magic('store' , '-r')
|
|
1169
|
ip.run_line_magic('store' , '-r')
|
|
1131
|
nt.assert_equal(ip.user_ns['var'], 39)
|
|
1170
|
nt.assert_equal(ip.user_ns['var'], 39)
|
|
1132
|
|
|
1171
|
|
|
1133
|
|
|
1172
|
|
|
1134
|
def _run_edit_test(arg_s, exp_filename=None,
|
|
1173
|
def _run_edit_test(arg_s, exp_filename=None,
|
|
1135
|
exp_lineno=-1,
|
|
1174
|
exp_lineno=-1,
|
|
1136
|
exp_contents=None,
|
|
1175
|
exp_contents=None,
|
|
1137
|
exp_is_temp=None):
|
|
1176
|
exp_is_temp=None):
|
|
1138
|
ip = get_ipython()
|
|
1177
|
ip = get_ipython()
|
|
1139
|
M = code.CodeMagics(ip)
|
|
1178
|
M = code.CodeMagics(ip)
|
|
1140
|
last_call = ['','']
|
|
1179
|
last_call = ['','']
|
|
1141
|
opts,args = M.parse_options(arg_s,'prxn:')
|
|
1180
|
opts,args = M.parse_options(arg_s,'prxn:')
|
|
1142
|
filename, lineno, is_temp = M._find_edit_target(ip, args, opts, last_call)
|
|
1181
|
filename, lineno, is_temp = M._find_edit_target(ip, args, opts, last_call)
|
|
1143
|
|
|
1182
|
|
|
1144
|
if exp_filename is not None:
|
|
1183
|
if exp_filename is not None:
|
|
1145
|
nt.assert_equal(exp_filename, filename)
|
|
1184
|
nt.assert_equal(exp_filename, filename)
|
|
1146
|
if exp_contents is not None:
|
|
1185
|
if exp_contents is not None:
|
|
1147
|
with io.open(filename, 'r', encoding='utf-8') as f:
|
|
1186
|
with io.open(filename, 'r', encoding='utf-8') as f:
|
|
1148
|
contents = f.read()
|
|
1187
|
contents = f.read()
|
|
1149
|
nt.assert_equal(exp_contents, contents)
|
|
1188
|
nt.assert_equal(exp_contents, contents)
|
|
1150
|
if exp_lineno != -1:
|
|
1189
|
if exp_lineno != -1:
|
|
1151
|
nt.assert_equal(exp_lineno, lineno)
|
|
1190
|
nt.assert_equal(exp_lineno, lineno)
|
|
1152
|
if exp_is_temp is not None:
|
|
1191
|
if exp_is_temp is not None:
|
|
1153
|
nt.assert_equal(exp_is_temp, is_temp)
|
|
1192
|
nt.assert_equal(exp_is_temp, is_temp)
|
|
1154
|
|
|
1193
|
|
|
1155
|
|
|
1194
|
|
|
1156
|
def test_edit_interactive():
|
|
1195
|
def test_edit_interactive():
|
|
1157
|
"""%edit on interactively defined objects"""
|
|
1196
|
"""%edit on interactively defined objects"""
|
|
1158
|
ip = get_ipython()
|
|
1197
|
ip = get_ipython()
|
|
1159
|
n = ip.execution_count
|
|
1198
|
n = ip.execution_count
|
|
1160
|
ip.run_cell(u"def foo(): return 1", store_history=True)
|
|
1199
|
ip.run_cell(u"def foo(): return 1", store_history=True)
|
|
1161
|
|
|
1200
|
|
|
1162
|
try:
|
|
1201
|
try:
|
|
1163
|
_run_edit_test("foo")
|
|
1202
|
_run_edit_test("foo")
|
|
1164
|
except code.InteractivelyDefined as e:
|
|
1203
|
except code.InteractivelyDefined as e:
|
|
1165
|
nt.assert_equal(e.index, n)
|
|
1204
|
nt.assert_equal(e.index, n)
|
|
1166
|
else:
|
|
1205
|
else:
|
|
1167
|
raise AssertionError("Should have raised InteractivelyDefined")
|
|
1206
|
raise AssertionError("Should have raised InteractivelyDefined")
|
|
1168
|
|
|
1207
|
|
|
1169
|
|
|
1208
|
|
|
1170
|
def test_edit_cell():
|
|
1209
|
def test_edit_cell():
|
|
1171
|
"""%edit [cell id]"""
|
|
1210
|
"""%edit [cell id]"""
|
|
1172
|
ip = get_ipython()
|
|
1211
|
ip = get_ipython()
|
|
1173
|
|
|
1212
|
|
|
1174
|
ip.run_cell(u"def foo(): return 1", store_history=True)
|
|
1213
|
ip.run_cell(u"def foo(): return 1", store_history=True)
|
|
1175
|
|
|
1214
|
|
|
1176
|
# test
|
|
1215
|
# test
|
|
1177
|
_run_edit_test("1", exp_contents=ip.user_ns['In'][1], exp_is_temp=True)
|
|
1216
|
_run_edit_test("1", exp_contents=ip.user_ns['In'][1], exp_is_temp=True)
|
|
1178
|
|
|
1217
|
|
|
1179
|
def test_edit_fname():
|
|
1218
|
def test_edit_fname():
|
|
1180
|
"""%edit file"""
|
|
1219
|
"""%edit file"""
|
|
1181
|
# test
|
|
1220
|
# test
|
|
1182
|
_run_edit_test("test file.py", exp_filename="test file.py")
|
|
1221
|
_run_edit_test("test file.py", exp_filename="test file.py")
|
|
1183
|
|
|
1222
|
|
|
1184
|
def test_bookmark():
|
|
1223
|
def test_bookmark():
|
|
1185
|
ip = get_ipython()
|
|
1224
|
ip = get_ipython()
|
|
1186
|
ip.run_line_magic('bookmark', 'bmname')
|
|
1225
|
ip.run_line_magic('bookmark', 'bmname')
|
|
1187
|
with tt.AssertPrints('bmname'):
|
|
1226
|
with tt.AssertPrints('bmname'):
|
|
1188
|
ip.run_line_magic('bookmark', '-l')
|
|
1227
|
ip.run_line_magic('bookmark', '-l')
|
|
1189
|
ip.run_line_magic('bookmark', '-d bmname')
|
|
1228
|
ip.run_line_magic('bookmark', '-d bmname')
|
|
1190
|
|
|
1229
|
|
|
1191
|
def test_ls_magic():
|
|
1230
|
def test_ls_magic():
|
|
1192
|
ip = get_ipython()
|
|
1231
|
ip = get_ipython()
|
|
1193
|
json_formatter = ip.display_formatter.formatters['application/json']
|
|
1232
|
json_formatter = ip.display_formatter.formatters['application/json']
|
|
1194
|
json_formatter.enabled = True
|
|
1233
|
json_formatter.enabled = True
|
|
1195
|
lsmagic = ip.magic('lsmagic')
|
|
1234
|
lsmagic = ip.magic('lsmagic')
|
|
1196
|
with warnings.catch_warnings(record=True) as w:
|
|
1235
|
with warnings.catch_warnings(record=True) as w:
|
|
1197
|
j = json_formatter(lsmagic)
|
|
1236
|
j = json_formatter(lsmagic)
|
|
1198
|
nt.assert_equal(sorted(j), ['cell', 'line'])
|
|
1237
|
nt.assert_equal(sorted(j), ['cell', 'line'])
|
|
1199
|
nt.assert_equal(w, []) # no warnings
|
|
1238
|
nt.assert_equal(w, []) # no warnings
|
|
1200
|
|
|
1239
|
|
|
1201
|
def test_strip_initial_indent():
|
|
1240
|
def test_strip_initial_indent():
|
|
1202
|
def sii(s):
|
|
1241
|
def sii(s):
|
|
1203
|
lines = s.splitlines()
|
|
1242
|
lines = s.splitlines()
|
|
1204
|
return '\n'.join(code.strip_initial_indent(lines))
|
|
1243
|
return '\n'.join(code.strip_initial_indent(lines))
|
|
1205
|
|
|
1244
|
|
|
1206
|
nt.assert_equal(sii(" a = 1\nb = 2"), "a = 1\nb = 2")
|
|
1245
|
nt.assert_equal(sii(" a = 1\nb = 2"), "a = 1\nb = 2")
|
|
1207
|
nt.assert_equal(sii(" a\n b\nc"), "a\n b\nc")
|
|
1246
|
nt.assert_equal(sii(" a\n b\nc"), "a\n b\nc")
|
|
1208
|
nt.assert_equal(sii("a\n b"), "a\n b")
|
|
1247
|
nt.assert_equal(sii("a\n b"), "a\n b")
|
|
1209
|
|
|
1248
|
|
|
1210
|
def test_logging_magic_quiet_from_arg():
|
|
1249
|
def test_logging_magic_quiet_from_arg():
|
|
1211
|
_ip.config.LoggingMagics.quiet = False
|
|
1250
|
_ip.config.LoggingMagics.quiet = False
|
|
1212
|
lm = logging.LoggingMagics(shell=_ip)
|
|
1251
|
lm = logging.LoggingMagics(shell=_ip)
|
|
1213
|
with TemporaryDirectory() as td:
|
|
1252
|
with TemporaryDirectory() as td:
|
|
1214
|
try:
|
|
1253
|
try:
|
|
1215
|
with tt.AssertNotPrints(re.compile("Activating.*")):
|
|
1254
|
with tt.AssertNotPrints(re.compile("Activating.*")):
|
|
1216
|
lm.logstart('-q {}'.format(
|
|
1255
|
lm.logstart('-q {}'.format(
|
|
1217
|
os.path.join(td, "quiet_from_arg.log")))
|
|
1256
|
os.path.join(td, "quiet_from_arg.log")))
|
|
1218
|
finally:
|
|
1257
|
finally:
|
|
1219
|
_ip.logger.logstop()
|
|
1258
|
_ip.logger.logstop()
|
|
1220
|
|
|
1259
|
|
|
1221
|
def test_logging_magic_quiet_from_config():
|
|
1260
|
def test_logging_magic_quiet_from_config():
|
|
1222
|
_ip.config.LoggingMagics.quiet = True
|
|
1261
|
_ip.config.LoggingMagics.quiet = True
|
|
1223
|
lm = logging.LoggingMagics(shell=_ip)
|
|
1262
|
lm = logging.LoggingMagics(shell=_ip)
|
|
1224
|
with TemporaryDirectory() as td:
|
|
1263
|
with TemporaryDirectory() as td:
|
|
1225
|
try:
|
|
1264
|
try:
|
|
1226
|
with tt.AssertNotPrints(re.compile("Activating.*")):
|
|
1265
|
with tt.AssertNotPrints(re.compile("Activating.*")):
|
|
1227
|
lm.logstart(os.path.join(td, "quiet_from_config.log"))
|
|
1266
|
lm.logstart(os.path.join(td, "quiet_from_config.log"))
|
|
1228
|
finally:
|
|
1267
|
finally:
|
|
1229
|
_ip.logger.logstop()
|
|
1268
|
_ip.logger.logstop()
|
|
1230
|
|
|
1269
|
|
|
1231
|
|
|
1270
|
|
|
1232
|
def test_logging_magic_not_quiet():
|
|
1271
|
def test_logging_magic_not_quiet():
|
|
1233
|
_ip.config.LoggingMagics.quiet = False
|
|
1272
|
_ip.config.LoggingMagics.quiet = False
|
|
1234
|
lm = logging.LoggingMagics(shell=_ip)
|
|
1273
|
lm = logging.LoggingMagics(shell=_ip)
|
|
1235
|
with TemporaryDirectory() as td:
|
|
1274
|
with TemporaryDirectory() as td:
|
|
1236
|
try:
|
|
1275
|
try:
|
|
1237
|
with tt.AssertPrints(re.compile("Activating.*")):
|
|
1276
|
with tt.AssertPrints(re.compile("Activating.*")):
|
|
1238
|
lm.logstart(os.path.join(td, "not_quiet.log"))
|
|
1277
|
lm.logstart(os.path.join(td, "not_quiet.log"))
|
|
1239
|
finally:
|
|
1278
|
finally:
|
|
1240
|
_ip.logger.logstop()
|
|
1279
|
_ip.logger.logstop()
|
|
1241
|
|
|
1280
|
|
|
1242
|
|
|
1281
|
|
|
1243
|
def test_time_no_var_expand():
|
|
1282
|
def test_time_no_var_expand():
|
|
1244
|
_ip.user_ns['a'] = 5
|
|
1283
|
_ip.user_ns['a'] = 5
|
|
1245
|
_ip.user_ns['b'] = []
|
|
1284
|
_ip.user_ns['b'] = []
|
|
1246
|
_ip.magic('time b.append("{a}")')
|
|
1285
|
_ip.magic('time b.append("{a}")')
|
|
1247
|
assert _ip.user_ns['b'] == ['{a}']
|
|
1286
|
assert _ip.user_ns['b'] == ['{a}']
|
|
1248
|
|
|
1287
|
|
|
1249
|
|
|
1288
|
|
|
1250
|
# this is slow, put at the end for local testing.
|
|
1289
|
# this is slow, put at the end for local testing.
|
|
1251
|
def test_timeit_arguments():
|
|
1290
|
def test_timeit_arguments():
|
|
1252
|
"Test valid timeit arguments, should not cause SyntaxError (GH #1269)"
|
|
1291
|
"Test valid timeit arguments, should not cause SyntaxError (GH #1269)"
|
|
1253
|
if sys.version_info < (3,7):
|
|
1292
|
if sys.version_info < (3,7):
|
|
1254
|
_ip.magic("timeit -n1 -r1 ('#')")
|
|
1293
|
_ip.magic("timeit -n1 -r1 ('#')")
|
|
1255
|
else:
|
|
1294
|
else:
|
|
1256
|
# 3.7 optimize no-op statement like above out, and complain there is
|
|
1295
|
# 3.7 optimize no-op statement like above out, and complain there is
|
|
1257
|
# nothing in the for loop.
|
|
1296
|
# nothing in the for loop.
|
|
1258
|
_ip.magic("timeit -n1 -r1 a=('#')")
|
|
1297
|
_ip.magic("timeit -n1 -r1 a=('#')")
|
|
1259
|
|
|
1298
|
|
|
1260
|
|
|
1299
|
|
|
1261
|
TEST_MODULE = """
|
|
1300
|
TEST_MODULE = """
|
|
1262
|
print('Loaded my_tmp')
|
|
1301
|
print('Loaded my_tmp')
|
|
1263
|
if __name__ == "__main__":
|
|
1302
|
if __name__ == "__main__":
|
|
1264
|
print('I just ran a script')
|
|
1303
|
print('I just ran a script')
|
|
1265
|
"""
|
|
1304
|
"""
|
|
1266
|
|
|
1305
|
|
|
1267
|
|
|
1306
|
|
|
1268
|
def test_run_module_from_import_hook():
|
|
1307
|
def test_run_module_from_import_hook():
|
|
1269
|
"Test that a module can be loaded via an import hook"
|
|
1308
|
"Test that a module can be loaded via an import hook"
|
|
1270
|
with TemporaryDirectory() as tmpdir:
|
|
1309
|
with TemporaryDirectory() as tmpdir:
|
|
1271
|
fullpath = os.path.join(tmpdir, 'my_tmp.py')
|
|
1310
|
fullpath = os.path.join(tmpdir, 'my_tmp.py')
|
|
1272
|
Path(fullpath).write_text(TEST_MODULE)
|
|
1311
|
Path(fullpath).write_text(TEST_MODULE)
|
|
1273
|
|
|
1312
|
|
|
1274
|
class MyTempImporter(object):
|
|
1313
|
class MyTempImporter(object):
|
|
1275
|
def __init__(self):
|
|
1314
|
def __init__(self):
|
|
1276
|
pass
|
|
1315
|
pass
|
|
1277
|
|
|
1316
|
|
|
1278
|
def find_module(self, fullname, path=None):
|
|
1317
|
def find_module(self, fullname, path=None):
|
|
1279
|
if 'my_tmp' in fullname:
|
|
1318
|
if 'my_tmp' in fullname:
|
|
1280
|
return self
|
|
1319
|
return self
|
|
1281
|
return None
|
|
1320
|
return None
|
|
1282
|
|
|
1321
|
|
|
1283
|
def load_module(self, name):
|
|
1322
|
def load_module(self, name):
|
|
1284
|
import imp
|
|
1323
|
import imp
|
|
1285
|
return imp.load_source('my_tmp', fullpath)
|
|
1324
|
return imp.load_source('my_tmp', fullpath)
|
|
1286
|
|
|
1325
|
|
|
1287
|
def get_code(self, fullname):
|
|
1326
|
def get_code(self, fullname):
|
|
1288
|
return compile(Path(fullpath).read_text(), "foo", "exec")
|
|
1327
|
return compile(Path(fullpath).read_text(), "foo", "exec")
|
|
1289
|
|
|
1328
|
|
|
1290
|
def is_package(self, __):
|
|
1329
|
def is_package(self, __):
|
|
1291
|
return False
|
|
1330
|
return False
|
|
1292
|
|
|
1331
|
|
|
1293
|
sys.meta_path.insert(0, MyTempImporter())
|
|
1332
|
sys.meta_path.insert(0, MyTempImporter())
|
|
1294
|
|
|
1333
|
|
|
1295
|
with capture_output() as captured:
|
|
1334
|
with capture_output() as captured:
|
|
1296
|
_ip.magic("run -m my_tmp")
|
|
1335
|
_ip.magic("run -m my_tmp")
|
|
1297
|
_ip.run_cell("import my_tmp")
|
|
1336
|
_ip.run_cell("import my_tmp")
|
|
1298
|
|
|
1337
|
|
|
1299
|
output = "Loaded my_tmp\nI just ran a script\nLoaded my_tmp\n"
|
|
1338
|
output = "Loaded my_tmp\nI just ran a script\nLoaded my_tmp\n"
|
|
1300
|
nt.assert_equal(output, captured.stdout)
|
|
1339
|
nt.assert_equal(output, captured.stdout)
|
|
1301
|
|
|
1340
|
|
|
1302
|
sys.meta_path.pop(0)
|
|
1341
|
sys.meta_path.pop(0)
|