##// END OF EJS Templates
fix event names
fix event names

File last commit:

r11124:9567c77a
r11226:c4ac215c
Show More
test_magic.py
866 lines | 24.8 KiB | text/x-python | PythonLexer
MinRK
test and fix %notebook magic
r6211 # -*- coding: utf-8 -*-
Fernando Perez
Update decorators and test scripts.
r1848 """Tests for various magic functions.
Ville M. Vainio
add test_magic, with single test (for rehashx)
r1735
Fernando Perez
Update decorators and test scripts.
r1848 Needs to be run by nose (to make ipython session available).
Ville M. Vainio
add test_magic, with single test (for rehashx)
r1735 """
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 from __future__ import absolute_import
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
MinRK
test and fix %notebook magic
r6211 import io
Fernando Perez
Update decorators and test scripts.
r1848 import os
MinRK
add failing test for %tb after SyntaxError
r6108 import sys
from StringIO import StringIO
Fernando Perez
Remove next_input nonsense in magic calls (but keep functionality).
r6974 from unittest import TestCase
Fernando Perez
Update decorators and test scripts.
r1848
Thomas Kluyver
Call invalidate_caches() when testing installing extension.
r7019 try:
from importlib import invalidate_caches # Required from Python 3.3
except ImportError:
def invalidate_caches():
pass
Fernando Perez
Update decorators and test scripts.
r1848 import nose.tools as nt
Fernando Perez
Fix various test failures from then new magics API.
r6942 from IPython.core import magic
Fernando Perez
Remove next_input nonsense in magic calls (but keep functionality).
r6974 from IPython.core.magic import (Magics, magics_class, line_magic,
cell_magic, line_cell_magic,
register_line_magic, register_cell_magic,
register_line_cell_magic)
MinRK
test %edit magic on interactively defined objects
r8993 from IPython.core.magics import execution, script, code
MinRK
test and fix %notebook magic
r6211 from IPython.nbformat.v3.tests.nbexamples import nb0
from IPython.nbformat import current
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762 from IPython.testing import decorators as dec
Fernando Perez
Ensure that we don't damage the __builtins__ object after %run....
r1955 from IPython.testing import tools as tt
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 from IPython.utils import py3compat
MinRK
exercise `%magic` in tests
r10996 from IPython.utils.io import capture_output
Thomas Kluyver
Add test for installing, loading, unloading an extension.
r6183 from IPython.utils.tempdir import TemporaryDirectory
MinRK
add basic tests for %%file, %%script
r7407 from IPython.utils.process import find_cmd
Fernando Perez
Ensure that we don't damage the __builtins__ object after %run....
r1955
Fernando Perez
Update decorators and test scripts.
r1848 #-----------------------------------------------------------------------------
# Test functions begin
Fernando Perez
Massive amount of work to improve the test suite, restores doctests....
r2414 #-----------------------------------------------------------------------------
Fernando Perez
Move tests for magics that are terminal-specific to their own file.
r5436
Fernando Perez
Renamed @register_magics to @magics_class to avoid confusion....
r6973 @magic.magics_class
Fernando Perez
Fix a few more test failures from magic API changes.
r6943 class DummyMagics(magic.Magics): pass
MinRK
test and fix %notebook magic
r6211
Ville M. Vainio
add test_magic, with single test (for rehashx)
r1735 def test_rehashx():
# clear up everything
Brian Granger
Massive refactoring of of the core....
r2245 _ip = get_ipython()
_ip.alias_manager.alias_table.clear()
Ville M. Vainio
add test_magic, with single test (for rehashx)
r1735 del _ip.db['syscmdlist']
_ip.magic('rehashx')
# Practically ALL ipython development systems will have more than 10 aliases
MinRK
remove yields from test_magic...
r11060 nt.assert_true(len(_ip.alias_manager.alias_table) > 10)
Thomas Kluyver
Replacing some .items() calls with .iteritems() for cleaner conversion with 2to3.
r3114 for key, val in _ip.alias_manager.alias_table.iteritems():
Ville M. Vainio
add test_magic, with single test (for rehashx)
r1735 # we must strip dots from alias names
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_not_in('.', key)
Ville M. Vainio
add test_magic, with single test (for rehashx)
r1735
# rehashx must fill up syscmdlist
scoms = _ip.db['syscmdlist']
MinRK
remove yields from test_magic...
r11060 nt.assert_true(len(scoms) > 10)
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762
Fernando Perez
Fix argument parsing bug in win32, clean up temp files in %hist doctest.
r2450 def test_magic_parse_options():
"""Test that we don't mangle paths when parsing magic options."""
ip = get_ipython()
path = 'c:\\x'
Fernando Perez
Fix a few more test failures from magic API changes.
r6943 m = DummyMagics(ip)
Fernando Perez
Fix various test failures from then new magics API.
r6942 opts = m.parse_options('-f %s' % path,'f:')[0]
Fernando Perez
Fix argument parsing bug in win32, clean up temp files in %hist doctest.
r2450 # argv splitting is os-dependent
if os.name == 'posix':
expected = 'c:x'
else:
expected = path
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(opts['f'], expected)
Fernando Perez
Fix argument parsing bug in win32, clean up temp files in %hist doctest.
r2450
MinRK
test Magic.parse_options with long options
r7041 def test_magic_parse_long_options():
"""Magic.parse_options can handle --foo=bar long options"""
ip = get_ipython()
m = DummyMagics(ip)
opts, _ = m.parse_options('--foo --bar=bubble', 'a', 'foo', 'bar=')
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_in('foo', opts)
nt.assert_in('bar', opts)
nt.assert_equal(opts['bar'], "bubble")
MinRK
test Magic.parse_options with long options
r7041
MinRK
Allow IPython to run without sqlite3...
r5147
@dec.skip_without('sqlite3')
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762 def doctest_hist_f():
"""Test %hist -f with temporary filename.
In [9]: import tempfile
In [10]: tfile = tempfile.mktemp('.py','tmp-ipython-')
Thomas Kluyver
Passing IPython.core tests.
r3396 In [11]: %hist -nl -f $tfile 3
Fernando Perez
Fix argument parsing bug in win32, clean up temp files in %hist doctest.
r2450
In [13]: import os; os.unlink(tfile)
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762 """
MinRK
Allow IPython to run without sqlite3...
r5147 @dec.skip_without('sqlite3')
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762 def doctest_hist_r():
"""Test %hist -r
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 XXX - This test is not recording the output correctly. For some reason, in
testing mode the raw history isn't getting populated. No idea why.
Disabling the output checking for now, though at least we do run it.
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 In [1]: 'hist' in _ip.lsmagic()
Out[1]: True
Fernando Perez
Fix some tests using magics....
r2092
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 In [2]: x=1
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762
Thomas Kluyver
Passing IPython.core tests.
r3396 In [3]: %hist -rl 2
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 x=1 # random
%hist -r 2
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762 """
MinRK
Allow IPython to run without sqlite3...
r5147
@dec.skip_without('sqlite3')
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 def doctest_hist_op():
"""Test %hist -op
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 In [1]: class b(float):
...: pass
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 ...:
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 In [2]: class s(object):
...: def __str__(self):
...: return 's'
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 ...:
In [3]:
In [4]: class r(b):
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 ...: def __repr__(self):
...: return 'r'
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 ...:
In [5]: class sr(s,r): pass
...:
In [6]:
In [7]: bb=b()
In [8]: ss=s()
In [9]: rr=r()
In [10]: ssrr=sr()
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 In [11]: 4.5
Out[11]: 4.5
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 In [12]: str(ss)
Out[12]: 's'
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441
In [13]:
In [14]: %hist -op
>>> class b:
... pass
...
>>> class s(b):
... def __str__(self):
... return 's'
...
>>>
>>> class r(b):
... def __repr__(self):
... return 'r'
...
>>> class sr(s,r): pass
>>>
>>> bb=b()
>>> ss=s()
>>> rr=r()
>>> ssrr=sr()
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 >>> 4.5
4.5
>>> str(ss)
's'
Fernando Perez
Changed %hist to default to NOT printing numbers, added -p and -o options....
r2441 >>>
"""
MinRK
Allow IPython to run without sqlite3...
r5147
@dec.skip_without('sqlite3')
Thomas Kluyver
Add test for magic macro command.
r3385 def test_macro():
ip = get_ipython()
ip.history_manager.reset() # Clear any existing history.
cmds = ["a=1", "def b():\n return a**2", "print(a,b())"]
Thomas Kluyver
Passing IPython.core tests.
r3396 for i, cmd in enumerate(cmds, start=1):
ip.history_manager.store_inputs(i, cmd)
Thomas Kluyver
Add test for magic macro command.
r3385 ip.magic("macro test 1-3")
nt.assert_equal(ip.user_ns["test"].value, "\n".join(cmds)+"\n")
MinRK
cleanup nt.assert usage in test_magic...
r11061 # List macros
nt.assert_in("test", ip.magic("macro"))
Fernando Perez
Progress towards getting the test suite in shape again....
r2392
MinRK
Allow IPython to run without sqlite3...
r5147
@dec.skip_without('sqlite3')
Thomas Kluyver
Refactor execution logic, so that a multi-line macro is correctly expanded and executed (+ unit test). History is now stored after all input transformations have been done.
r3414 def test_macro_run():
"""Test that we can run a multi-line macro successfully."""
ip = get_ipython()
ip.history_manager.reset()
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 cmds = ["a=10", "a+=1", py3compat.doctest_refactor_print("print a"),
"%macro test 2-3"]
Thomas Kluyver
Refactor execution logic, so that a multi-line macro is correctly expanded and executed (+ unit test). History is now stored after all input transformations have been done.
r3414 for cmd in cmds:
Thomas Kluyver
Change run_cell to not store history by default.
r4995 ip.run_cell(cmd, store_history=True)
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 nt.assert_equal(ip.user_ns["test"].value,
py3compat.doctest_refactor_print("a+=1\nprint a\n"))
Thomas Kluyver
Use AssertPrints in test_magic.
r4903 with tt.AssertPrints("12"):
Thomas Kluyver
Refactor execution logic, so that a multi-line macro is correctly expanded and executed (+ unit test). History is now stored after all input transformations have been done.
r3414 ip.run_cell("test")
Thomas Kluyver
Use AssertPrints in test_magic.
r4903 with tt.AssertPrints("13"):
Thomas Kluyver
Refactor execution logic, so that a multi-line macro is correctly expanded and executed (+ unit test). History is now stored after all input transformations have been done.
r3414 ip.run_cell("test")
Fernando Perez
Fix some tests using magics....
r2092
MinRK
exercise `%magic` in tests
r10996 def test_magic_magic():
"""Test %magic"""
ip = get_ipython()
with capture_output() as captured:
ip.magic("magic")
stdout = captured.stdout
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_in('%magic', stdout)
nt.assert_in('IPython', stdout)
nt.assert_in('Available', stdout)
MinRK
exercise `%magic` in tests
r10996
Paul Ivanov
tests for %clear extension
r5939 @dec.skipif_not_numpy
Paul Ivanov
%reset now takes optional in/out/dhist/array args...
r5965 def test_numpy_reset_array_undec():
"Test '%reset array' functionality"
Fernando Perez
Update decorators and test scripts.
r1848 _ip.ex('import numpy as np')
_ip.ex('a = np.empty(2)')
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_in('a', _ip.user_ns)
Paul Ivanov
%reset now takes optional in/out/dhist/array args...
r5965 _ip.magic('reset -f array')
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_not_in('a', _ip.user_ns)
Paul Ivanov
tests for %clear extension
r5939
Paul Ivanov
separate %reset tests for finer granularity
r5976 def test_reset_out():
"Test '%reset out' magic"
Paul Ivanov
tests for %clear extension
r5939 _ip.run_cell("parrot = 'dead'", store_history=True)
Paul Ivanov
%reset now takes optional in/out/dhist/array args...
r5965 # test '%reset -f out', make an Out prompt
Paul Ivanov
tests for %clear extension
r5939 _ip.run_cell("parrot", store_history=True)
nt.assert_true('dead' in [_ip.user_ns[x] for x in '_','__','___'])
Paul Ivanov
%reset now takes optional in/out/dhist/array args...
r5965 _ip.magic('reset -f out')
Paul Ivanov
tests for %clear extension
r5939 nt.assert_false('dead' in [_ip.user_ns[x] for x in '_','__','___'])
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_equal(len(_ip.user_ns['Out']), 0)
Paul Ivanov
tests for %clear extension
r5939
Paul Ivanov
separate %reset tests for finer granularity
r5976 def test_reset_in():
"Test '%reset in' magic"
Paul Ivanov
%reset now takes optional in/out/dhist/array args...
r5965 # test '%reset -f in'
Paul Ivanov
tests for %clear extension
r5939 _ip.run_cell("parrot", store_history=True)
nt.assert_true('parrot' in [_ip.user_ns[x] for x in '_i','_ii','_iii'])
Paul Ivanov
%reset now takes optional in/out/dhist/array args...
r5965 _ip.magic('%reset -f in')
Paul Ivanov
tests for %clear extension
r5939 nt.assert_false('parrot' in [_ip.user_ns[x] for x in '_i','_ii','_iii'])
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_equal(len(set(_ip.user_ns['In'])), 1)
Paul Ivanov
tests for %clear extension
r5939
Paul Ivanov
separate %reset tests for finer granularity
r5976 def test_reset_dhist():
"Test '%reset dhist' magic"
Paul Ivanov
tests for %clear extension
r5939 _ip.run_cell("tmp = [d for d in _dh]") # copy before clearing
Paul Ivanov
separate %reset tests for finer granularity
r5976 _ip.magic('cd ' + os.path.dirname(nt.__file__))
Paul Ivanov
tests for %clear extension
r5939 _ip.magic('cd -')
nt.assert_true(len(_ip.user_ns['_dh']) > 0)
Paul Ivanov
%reset now takes optional in/out/dhist/array args...
r5965 _ip.magic('reset -f dhist')
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_equal(len(_ip.user_ns['_dh']), 0)
Paul Ivanov
tests for %clear extension
r5939 _ip.run_cell("_dh = [d for d in tmp]") #restore
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762
Paul Ivanov
separate %reset tests for finer granularity
r5976 def test_reset_in_length():
"Test that '%reset in' preserves In[] length"
Paul Ivanov
clear out the contents of but keep length of In[]
r5957 _ip.run_cell("print 'foo'")
Paul Ivanov
%reset now takes optional in/out/dhist/array args...
r5965 _ip.run_cell("reset -f in")
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_equal(len(_ip.user_ns['In']), _ip.displayhook.prompt_count+1)
Fernando Perez
Fix a number of bugs with %history, add proper tests....
r1762
MinRK
add failing test for %tb after SyntaxError
r6108 def test_tb_syntaxerror():
"""test %tb after a SyntaxError"""
ip = get_ipython()
ip.run_cell("for")
# trap and validate stdout
save_stdout = sys.stdout
try:
sys.stdout = StringIO()
ip.run_cell("%tb")
out = sys.stdout.getvalue()
finally:
sys.stdout = save_stdout
# trim output, and only check the last line
last_line = out.rstrip().splitlines()[-1].strip()
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(last_line, "SyntaxError: invalid syntax")
MinRK
add failing test for %tb after SyntaxError
r6108
Fernando Perez
Fix broken %time magic....
r2411
Thomas Kluyver
Remove code from prefilter that duplicates functionality in inputsplitter.
r8216 def test_time():
ip = get_ipython()
Thomas Kluyver
Add %time within function to doctest.
r3478
Jan Schulz
time magic: shorten unnecessary output on windows...
r9728 with tt.AssertPrints("Wall time: "):
Thomas Kluyver
Remove code from prefilter that duplicates functionality in inputsplitter.
r8216 ip.run_cell("%time None")
ip.run_cell("def f(kmjy):\n"
" %time print (2*kmjy)")
Jan Schulz
time magic: shorten unnecessary output on windows...
r9728 with tt.AssertPrints("Wall time: "):
Thomas Kluyver
Remove code from prefilter that duplicates functionality in inputsplitter.
r8216 with tt.AssertPrints("hihi", suppress=False):
ip.run_cell("f('hi')")
Fernando Perez
Work around a bug in Python's shlex module with unicode input....
r2650
Jan Schulz
time magic: shorten unnecessary output on windows...
r9728
@dec.skip_win32
def test_time2():
ip = get_ipython()
with tt.AssertPrints("CPU times: user "):
ip.run_cell("%time None")
Thomas Kluyver
Failing test for gh-3334
r10671 def test_time3():
"""Erroneous magic function calls, issue gh-3334"""
ip = get_ipython()
ip.user_ns.pop('run', None)
with tt.AssertNotPrints("not found", channel='stderr'):
ip.run_cell("%%time\n"
"run = 0\n"
"run += 1")
Fernando Perez
Add transformers to understand code pasted with >>> or IPython prompts....
r2426 def test_doctest_mode():
"Toggle doctest_mode twice, it should be a no-op and run without error"
_ip.magic('doctest_mode')
_ip.magic('doctest_mode')
Fernando Perez
Work around a bug in Python's shlex module with unicode input....
r2650
def test_parse_options():
"""Tests for basic options parsing in magics."""
# These are only the most minimal of tests, more should be added later. At
# the very least we check that basic text/unicode calls work OK.
Fernando Perez
Fix a few more test failures from magic API changes.
r6943 m = DummyMagics(_ip)
Fernando Perez
Fix various test failures from then new magics API.
r6942 nt.assert_equal(m.parse_options('foo', '')[1], 'foo')
nt.assert_equal(m.parse_options(u'foo', '')[1], u'foo')
Fernando Perez
Work around a bug in Python's shlex module with unicode input....
r2650
Fernando Perez
Add transformers to understand code pasted with >>> or IPython prompts....
r2426
Fernando Perez
Work around a bug in Python's shlex module with unicode input....
r2650 def test_dirops():
"""Test various directory handling operations."""
Min RK
fix various tests on Windows...
r4105 # curpath = lambda :os.path.splitdrive(os.getcwdu())[1].replace('\\','/')
curpath = os.getcwdu
Thomas Kluyver
IPython behaves properly when started in a directory with non-ascii characters in the name. + Unit test....
r3451 startdir = os.getcwdu()
Gabriel
Fix tests to work when ~/.config/ipython contains a symlink.
r5663 ipdir = os.path.realpath(_ip.ipython_dir)
Fernando Perez
Work around a bug in Python's shlex module with unicode input....
r2650 try:
_ip.magic('cd "%s"' % ipdir)
nt.assert_equal(curpath(), ipdir)
_ip.magic('cd -')
nt.assert_equal(curpath(), startdir)
_ip.magic('pushd "%s"' % ipdir)
nt.assert_equal(curpath(), ipdir)
_ip.magic('popd')
nt.assert_equal(curpath(), startdir)
finally:
os.chdir(startdir)
Fernando Perez
Fixed unused %cpaste test and moved into proper test suite.
r2655
Fernando Perez
Fix small bug where %xmode without arguments failed to run....
r3146 def test_xmode():
# Calling xmode three times should be a no-op
xmode = _ip.InteractiveTB.mode
for i in range(3):
_ip.magic("xmode")
nt.assert_equal(_ip.InteractiveTB.mode, xmode)
Thomas Kluyver
Add test for hard reset.
r3522
def test_reset_hard():
monitor = []
class A(object):
def __del__(self):
monitor.append(1)
def __repr__(self):
return "<A instance>"
_ip.user_ns["a"] = A()
_ip.run_cell("a")
nt.assert_equal(monitor, [])
Fernando Perez
Fix remaining test failures....
r6916 _ip.magic("reset -f")
Thomas Kluyver
Add test for hard reset.
r3522 nt.assert_equal(monitor, [1])
Thomas Kluyver
Create test for %xdel magic. Not passing yet, although the equivalent works in interactive use.
r3824
class TestXdel(tt.TempFileMixin):
def test_xdel(self):
"""Test that references from %run are cleared by xdel."""
src = ("class A(object):\n"
" monitor = []\n"
" def __del__(self):\n"
" self.monitor.append(1)\n"
"a = A()\n")
self.mktmp(src)
Thomas Kluyver
Remove remaining references held by test suite, so that xdel test passes.
r3832 # %run creates some hidden references...
Thomas Kluyver
Create test for %xdel magic. Not passing yet, although the equivalent works in interactive use.
r3824 _ip.magic("run %s" % self.fname)
Thomas Kluyver
Remove remaining references held by test suite, so that xdel test passes.
r3832 # ... as does the displayhook.
Thomas Kluyver
Create test for %xdel magic. Not passing yet, although the equivalent works in interactive use.
r3824 _ip.run_cell("a")
Thomas Kluyver
Remove object references kept by the test suite in a better way.
r3844
Thomas Kluyver
Create test for %xdel magic. Not passing yet, although the equivalent works in interactive use.
r3824 monitor = _ip.user_ns["A"].monitor
nt.assert_equal(monitor, [])
Thomas Kluyver
Remove remaining references held by test suite, so that xdel test passes.
r3832
Thomas Kluyver
Remove object references kept by the test suite in a better way.
r3844 _ip.magic("xdel a")
Thomas Kluyver
Remove remaining references held by test suite, so that xdel test passes.
r3832
# Check that a's __del__ method has been called.
Thomas Kluyver
Create test for %xdel magic. Not passing yet, although the equivalent works in interactive use.
r3824 nt.assert_equal(monitor, [1])
MinRK
fix+test %who_ls type checking, skip %who doctests...
r3332
def doctest_who():
"""doctest for %who
In [1]: %reset -f
In [2]: alpha = 123
In [3]: beta = 'beta'
In [4]: %who int
alpha
In [5]: %who str
beta
In [6]: %whos
Variable Type Data/Info
----------------------------
alpha int 123
beta str beta
In [7]: %who_ls
Out[7]: ['alpha', 'beta']
MinRK
add `float_precision` trait to PlainTextFormatter...
r3350 """
Thomas Kluyver
Catch failure in repr() for %whos.
r6779 def test_whos():
"""Check that whos is protected against objects where repr() fails."""
class A(object):
def __repr__(self):
raise Exception()
_ip.user_ns['a'] = A()
_ip.magic("whos")
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 @py3compat.u_format
MinRK
add `float_precision` trait to PlainTextFormatter...
r3350 def doctest_precision():
"""doctest for %precision
Fernando Perez
Fix remaining test failures....
r6916 In [1]: f = get_ipython().display_formatter.formatters['text/plain']
MinRK
add `float_precision` trait to PlainTextFormatter...
r3350
In [2]: %precision 5
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 Out[2]: {u}'%.5f'
MinRK
add `float_precision` trait to PlainTextFormatter...
r3350
In [3]: f.float_format
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 Out[3]: {u}'%.5f'
MinRK
add `float_precision` trait to PlainTextFormatter...
r3350
In [4]: %precision %e
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 Out[4]: {u}'%e'
MinRK
add `float_precision` trait to PlainTextFormatter...
r3350
In [5]: f(3.1415927)
Thomas Kluyver
Fix various tests in IPython.core for Python 3.
r4895 Out[5]: {u}'3.141593e+00'
MinRK
add `float_precision` trait to PlainTextFormatter...
r3350 """
Thomas Kluyver
Add test for wildcard search syntax.
r5549 def test_psearch():
with tt.AssertPrints("dict.fromkeys"):
_ip.run_cell("dict.fr*?")
MinRK
add strict flag to arg_split, to optionally ignore shlex parse errors...
r5672 def test_timeit_shlex():
"""test shlex issues with timeit (#1109)"""
_ip.ex("def f(*a,**kw): pass")
_ip.magic('timeit -n1 "this is a bug".count(" ")')
_ip.magic('timeit -r1 -n1 f(" ", 1)')
_ip.magic('timeit -r1 -n1 f(" ", 1, " ", 2, " ")')
_ip.magic('timeit -r1 -n1 ("a " + "b")')
_ip.magic('timeit -r1 -n1 f("a " + "b")')
_ip.magic('timeit -r1 -n1 f("a " + "b ")')
Paul Ivanov
test for GH #1269
r5901
def test_timeit_arguments():
"Test valid timeit arguments, should not cause SyntaxError (GH #1269)"
_ip.magic("timeit ('#')")
Paul Ivanov
test for #1302
r5932
Fernando Perez
Fix various test failures from then new magics API.
r6942
Fernando Perez
Extend the previous test to cover both line and cell modes.
r7493 def test_timeit_special_syntax():
"Test %%timeit with IPython special syntax"
Fernando Perez
Add test for new timeit functionality.
r7492 @register_line_magic
def lmagic(line):
ip = get_ipython()
ip.user_ns['lmagic_out'] = line
Fernando Perez
Extend the previous test to cover both line and cell modes.
r7493 # line mode test
_ip.run_line_magic('timeit', '-n1 -r1 %lmagic my line')
Fernando Perez
Add test for new timeit functionality.
r7492 nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line')
Fernando Perez
Extend the previous test to cover both line and cell modes.
r7493 # cell mode test
_ip.run_cell_magic('timeit', '-n1 -r1', '%lmagic my line2')
nt.assert_equal(_ip.user_ns['lmagic_out'], 'my line2')
Fernando Perez
Add test for new timeit functionality.
r7492
Fernando Perez
Fix failing tests in core and zmq
r6970 @dec.skipif(execution.profile is None)
Paul Ivanov
test for #1302
r5932 def test_prun_quotes():
"Test that prun does not clobber string escapes (GH #1302)"
Jörgen Stenarson
Switch to raw string to match typing at prompt
r7458 _ip.magic(r"prun -q x = '\t'")
Paul Ivanov
test for #1302
r5932 nt.assert_equal(_ip.user_ns['x'], '\t')
Thomas Kluyver
Add test for installing, loading, unloading an extension.
r6183
def test_extension():
tmpdir = TemporaryDirectory()
orig_ipython_dir = _ip.ipython_dir
try:
_ip.ipython_dir = tmpdir.name
nt.assert_raises(ImportError, _ip.magic, "load_ext daft_extension")
url = os.path.join(os.path.dirname(__file__), "daft_extension.py")
_ip.magic("install_ext %s" % url)
_ip.user_ns.pop('arq', None)
Thomas Kluyver
Call invalidate_caches() when testing installing extension.
r7019 invalidate_caches() # Clear import caches
Thomas Kluyver
Add test for installing, loading, unloading an extension.
r6183 _ip.magic("load_ext daft_extension")
Bradley M. Froehle
Stop duplicating `nt.assert*` in `IPython.testing.tools`.
r7879 nt.assert_equal(_ip.user_ns['arq'], 185)
Thomas Kluyver
Add test for installing, loading, unloading an extension.
r6183 _ip.magic("unload_ext daft_extension")
assert 'arq' not in _ip.user_ns
finally:
_ip.ipython_dir = orig_ipython_dir
Bradley M. Froehle
Clean up TemporaryDirectory (prevent ResourceWarning)
r7805 tmpdir.cleanup()
Thomas Kluyver
Add test for installing, loading, unloading an extension.
r6183
MinRK
test and fix %notebook magic
r6211 def test_notebook_export_json():
with TemporaryDirectory() as td:
outfile = os.path.join(td, "nb.ipynb")
MinRK
unicode literals in notebook magic tests...
r6475 _ip.ex(py3compat.u_format(u"u = {u}'héllo'"))
MinRK
test and fix %notebook magic
r6211 _ip.magic("notebook -e %s" % outfile)
def test_notebook_export_py():
with TemporaryDirectory() as td:
outfile = os.path.join(td, "nb.py")
MinRK
unicode literals in notebook magic tests...
r6475 _ip.ex(py3compat.u_format(u"u = {u}'héllo'"))
MinRK
test and fix %notebook magic
r6211 _ip.magic("notebook -e %s" % outfile)
def test_notebook_reformat_py():
with TemporaryDirectory() as td:
infile = os.path.join(td, "nb.ipynb")
Thomas Kluyver
Specify encoding for io.open in notebook_reformat tests.
r6488 with io.open(infile, 'w', encoding='utf-8') as f:
MinRK
test and fix %notebook magic
r6211 current.write(nb0, f, 'json')
MinRK
unicode literals in notebook magic tests...
r6475 _ip.ex(py3compat.u_format(u"u = {u}'héllo'"))
MinRK
test and fix %notebook magic
r6211 _ip.magic("notebook -f py %s" % infile)
def test_notebook_reformat_json():
with TemporaryDirectory() as td:
infile = os.path.join(td, "nb.py")
Thomas Kluyver
Specify encoding for io.open in notebook_reformat tests.
r6488 with io.open(infile, 'w', encoding='utf-8') as f:
MinRK
test and fix %notebook magic
r6211 current.write(nb0, f, 'py')
MinRK
unicode literals in notebook magic tests...
r6475 _ip.ex(py3compat.u_format(u"u = {u}'héllo'"))
MinRK
test and fix %notebook magic
r6211 _ip.magic("notebook -f ipynb %s" % infile)
_ip.magic("notebook -f json %s" % infile)
Thomas Kluyver
Fix %env magic on Python 3....
r6489 def test_env():
env = _ip.magic("env")
assert isinstance(env, dict), type(env)
Fernando Perez
Add first set of cell magic tests.
r6975
class CellMagicTestCase(TestCase):
def check_ident(self, magic):
Fernando Perez
First implementation of cell magics that goes via inputsplitter....
r6976 # Manually called, we get the result
Fernando Perez
Rename a few methods as per review, also complete some docstrings.
r7003 out = _ip.run_cell_magic(magic, 'a', 'b')
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(out, ('a','b'))
Fernando Perez
First implementation of cell magics that goes via inputsplitter....
r6976 # Via run_cell, it goes into the user's namespace via displayhook
_ip.run_cell('%%' + magic +' c\nd')
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(_ip.user_ns['_'], ('c','d'))
Fernando Perez
Add first set of cell magic tests.
r6975
def test_cell_magic_func_deco(self):
"Cell magic using simple decorator"
@register_cell_magic
def cellm(line, cell):
return line, cell
self.check_ident('cellm')
def test_cell_magic_reg(self):
"Cell magic manually registered"
def cellm(line, cell):
return line, cell
_ip.register_magic_function(cellm, 'cell', 'cellm2')
self.check_ident('cellm2')
def test_cell_magic_class(self):
"Cell magics declared via a class"
@magics_class
class MyMagics(Magics):
@cell_magic
def cellm3(self, line, cell):
return line, cell
Fernando Perez
First implementation of cell magics that goes via inputsplitter....
r6976 _ip.register_magics(MyMagics)
self.check_ident('cellm3')
def test_cell_magic_class2(self):
"Cell magics declared via a class, #2"
@magics_class
class MyMagics2(Magics):
Fernando Perez
Add first set of cell magic tests.
r6975 @cell_magic('cellm4')
def cellm33(self, line, cell):
return line, cell
Fernando Perez
First implementation of cell magics that goes via inputsplitter....
r6976
_ip.register_magics(MyMagics2)
Fernando Perez
Add first set of cell magic tests.
r6975 self.check_ident('cellm4')
# Check that nothing is registered as 'cellm33'
c33 = _ip.find_cell_magic('cellm33')
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(c33, None)
MinRK
add basic tests for %%file, %%script
r7407
def test_file():
"""Basic %%file"""
ip = get_ipython()
with TemporaryDirectory() as td:
fname = os.path.join(td, 'file1')
ip.run_cell_magic("file", fname, u'\n'.join([
'line1',
'line2',
]))
with open(fname) as f:
s = f.read()
nt.assert_in('line1\n', s)
nt.assert_in('line2', s)
MinRK
test that var_expand is called on the line in %%file
r7903 def test_file_var_expand():
"""%%file $filename"""
ip = get_ipython()
with TemporaryDirectory() as td:
fname = os.path.join(td, 'file1')
ip.user_ns['filename'] = fname
ip.run_cell_magic("file", '$filename', u'\n'.join([
'line1',
'line2',
]))
with open(fname) as f:
s = f.read()
nt.assert_in('line1\n', s)
nt.assert_in('line2', s)
MinRK
add basic tests for %%file, %%script
r7407 def test_file_unicode():
"""%%file with unicode cell"""
ip = get_ipython()
with TemporaryDirectory() as td:
fname = os.path.join(td, 'file1')
ip.run_cell_magic("file", fname, u'\n'.join([
u'liné1',
u'liné2',
]))
with io.open(fname, encoding='utf-8') as f:
s = f.read()
nt.assert_in(u'liné1\n', s)
nt.assert_in(u'liné2', s)
def test_file_amend():
"""%%file -a amends files"""
ip = get_ipython()
with TemporaryDirectory() as td:
fname = os.path.join(td, 'file2')
ip.run_cell_magic("file", fname, u'\n'.join([
'line1',
'line2',
]))
ip.run_cell_magic("file", "-a %s" % fname, u'\n'.join([
'line3',
'line4',
]))
with open(fname) as f:
s = f.read()
nt.assert_in('line1\n', s)
nt.assert_in('line3\n', s)
Fernando Perez
Add support for finding cell magics with ?/??....
r6997
MinRK
add basic tests for %%file, %%script
r7407 def test_script_config():
ip = get_ipython()
ip.config.ScriptMagics.script_magics = ['whoda']
sm = script.ScriptMagics(shell=ip)
nt.assert_in('whoda', sm.magics['cell'])
@dec.skip_win32
def test_script_out():
ip = get_ipython()
ip.run_cell_magic("script", "--out output sh", "echo 'hi'")
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(ip.user_ns['output'], 'hi\n')
MinRK
add basic tests for %%file, %%script
r7407
@dec.skip_win32
def test_script_err():
ip = get_ipython()
ip.run_cell_magic("script", "--err error sh", "echo 'hello' >&2")
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(ip.user_ns['error'], 'hello\n')
MinRK
add basic tests for %%file, %%script
r7407
@dec.skip_win32
def test_script_out_err():
ip = get_ipython()
ip.run_cell_magic("script", "--out output --err error sh", "echo 'hi'\necho 'hello' >&2")
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(ip.user_ns['output'], 'hi\n')
nt.assert_equal(ip.user_ns['error'], 'hello\n')
MinRK
add basic tests for %%file, %%script
r7407
@dec.skip_win32
def test_script_bg_out():
ip = get_ipython()
ip.run_cell_magic("script", "--bg --out output sh", "echo 'hi'")
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(ip.user_ns['output'].read(), b'hi\n')
MinRK
add basic tests for %%file, %%script
r7407
@dec.skip_win32
def test_script_bg_err():
ip = get_ipython()
ip.run_cell_magic("script", "--bg --err error sh", "echo 'hello' >&2")
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(ip.user_ns['error'].read(), b'hello\n')
MinRK
add basic tests for %%file, %%script
r7407
@dec.skip_win32
def test_script_bg_out_err():
ip = get_ipython()
ip.run_cell_magic("script", "--bg --out output --err error sh", "echo 'hi'\necho 'hello' >&2")
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(ip.user_ns['output'].read(), b'hi\n')
nt.assert_equal(ip.user_ns['error'].read(), b'hello\n')
MinRK
add basic tests for %%file, %%script
r7407
def test_script_defaults():
ip = get_ipython()
for cmd in ['sh', 'bash', 'perl', 'ruby']:
try:
find_cmd(cmd)
except Exception:
pass
else:
nt.assert_in(cmd, ip.magics_manager.magics['cell'])
MinRK
add test for inspection of synonymous line/cell magics
r7436
@magics_class
class FooFoo(Magics):
"""class with both %foo and %%foo magics"""
@line_magic('foo')
def line_foo(self, line):
"I am line foo"
pass
Bradley M. Froehle
Add tests for %alias_magic.
r7712
MinRK
add test for inspection of synonymous line/cell magics
r7436 @cell_magic("foo")
def cell_foo(self, line, cell):
"I am cell foo, not line foo"
pass
def test_line_cell_info():
"""%%foo and %foo magics are distinguishable to inspect"""
ip = get_ipython()
ip.magics_manager.register(FooFoo)
oinfo = ip.object_inspect('foo')
nt.assert_true(oinfo['found'])
nt.assert_true(oinfo['ismagic'])
oinfo = ip.object_inspect('%%foo')
nt.assert_true(oinfo['found'])
nt.assert_true(oinfo['ismagic'])
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(oinfo['docstring'], FooFoo.cell_foo.__doc__)
MinRK
add test for inspection of synonymous line/cell magics
r7436
oinfo = ip.object_inspect('%foo')
nt.assert_true(oinfo['found'])
nt.assert_true(oinfo['ismagic'])
Bradley M. Froehle
s/nt.assert_equals/nt.assert_equal/
r7875 nt.assert_equal(oinfo['docstring'], FooFoo.line_foo.__doc__)
MinRK
add failing test for registering multiple instances of the same magic
r7583
def test_multiple_magics():
ip = get_ipython()
foo1 = FooFoo(ip)
foo2 = FooFoo(ip)
mm = ip.magics_manager
mm.register(foo1)
nt.assert_true(mm.magics['line']['foo'].im_self is foo1)
mm.register(foo2)
nt.assert_true(mm.magics['line']['foo'].im_self is foo2)
Bradley M. Froehle
Add tests for %alias_magic.
r7712
def test_alias_magic():
"""Test %alias_magic."""
ip = get_ipython()
mm = ip.magics_manager
# Basic operation: both cell and line magics are created, if possible.
ip.run_line_magic('alias_magic', 'timeit_alias timeit')
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_in('timeit_alias', mm.magics['line'])
nt.assert_in('timeit_alias', mm.magics['cell'])
Bradley M. Froehle
Add tests for %alias_magic.
r7712
# --cell is specified, line magic not created.
ip.run_line_magic('alias_magic', '--cell timeit_cell_alias timeit')
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_not_in('timeit_cell_alias', mm.magics['line'])
nt.assert_in('timeit_cell_alias', mm.magics['cell'])
Bradley M. Froehle
Add tests for %alias_magic.
r7712
# Test that line alias is created successfully.
ip.run_line_magic('alias_magic', '--line env_alias env')
nt.assert_equal(ip.run_line_magic('env', ''),
ip.run_line_magic('env_alias', ''))
Dominik Dabrowski
add test for %save
r7850
def test_save():
"""Test %save."""
ip = get_ipython()
ip.history_manager.reset() # Clear any existing history.
cmds = [u"a=1", u"def b():\n return a**2", u"print(a, b())"]
for i, cmd in enumerate(cmds, start=1):
ip.history_manager.store_inputs(i, cmd)
with TemporaryDirectory() as tmpdir:
file = os.path.join(tmpdir, "testsave.py")
ip.run_line_magic("save", "%s 1-10" % file)
with open(file) as f:
content = f.read()
nt.assert_equal(content.count(cmds[0]), 1)
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_in('coding: utf-8', content)
Dominik Dabrowski
add test for %save
r7850 ip.run_line_magic("save", "-a %s 1-10" % file)
with open(file) as f:
content = f.read()
nt.assert_equal(content.count(cmds[0]), 2)
MinRK
cleanup nt.assert usage in test_magic...
r11061 nt.assert_in('coding: utf-8', content)
Cavendish McKay
added test for %store, fixed storemagic
r8002
def test_store():
"""Test %store."""
ip = get_ipython()
ip.run_line_magic('load_ext', 'storemagic')
# make sure the storage is empty
ip.run_line_magic('store', '-z')
ip.user_ns['var'] = 42
ip.run_line_magic('store', 'var')
ip.user_ns['var'] = 39
ip.run_line_magic('store', '-r')
nt.assert_equal(ip.user_ns['var'], 42)
ip.run_line_magic('store', '-d var')
ip.user_ns['var'] = 39
ip.run_line_magic('store' , '-r')
nt.assert_equal(ip.user_ns['var'], 39)
MinRK
test %edit magic on interactively defined objects
r8993
def _run_edit_test(arg_s, exp_filename=None,
exp_lineno=-1,
exp_contents=None,
exp_is_temp=None):
ip = get_ipython()
M = code.CodeMagics(ip)
last_call = ['','']
opts,args = M.parse_options(arg_s,'prxn:')
filename, lineno, is_temp = M._find_edit_target(ip, args, opts, last_call)
if exp_filename is not None:
nt.assert_equal(exp_filename, filename)
if exp_contents is not None:
with io.open(filename, 'r') as f:
contents = f.read()
nt.assert_equal(exp_contents, contents)
if exp_lineno != -1:
nt.assert_equal(exp_lineno, lineno)
if exp_is_temp is not None:
nt.assert_equal(exp_is_temp, is_temp)
def test_edit_interactive():
"""%edit on interactively defined objects"""
ip = get_ipython()
n = ip.execution_count
ip.run_cell(u"def foo(): return 1", store_history=True)
try:
_run_edit_test("foo")
except code.InteractivelyDefined as e:
nt.assert_equal(e.index, n)
else:
Thomas Kluyver
Fix unrelated test....
r9141 raise AssertionError("Should have raised InteractivelyDefined")
MinRK
test %edit magic on interactively defined objects
r8993
def test_edit_cell():
"""%edit [cell id]"""
ip = get_ipython()
ip.run_cell(u"def foo(): return 1", store_history=True)
# test
_run_edit_test("1", exp_contents=ip.user_ns['In'][1], exp_is_temp=True)