test_run.py
203 lines
| 5.8 KiB
| text/x-python
|
PythonLexer
|
r2414 | """Tests for code execution (%run and related), which is particularly tricky. | ||
Because of how %run manages namespaces, and the fact that we are trying here to | ||||
verify subtle object deletion and reference counting issues, the %run tests | ||||
will be kept in this separate file. This makes it easier to aggregate in one | ||||
place the tricks needed to handle it; most other magics are much easier to test | ||||
and we do so in a common test_magic file. | ||||
""" | ||||
from __future__ import absolute_import | ||||
#----------------------------------------------------------------------------- | ||||
# Imports | ||||
#----------------------------------------------------------------------------- | ||||
import os | ||||
import sys | ||||
import tempfile | ||||
import nose.tools as nt | ||||
from IPython.testing import decorators as dec | ||||
from IPython.testing import tools as tt | ||||
#----------------------------------------------------------------------------- | ||||
# Test functions begin | ||||
#----------------------------------------------------------------------------- | ||||
def doctest_refbug(): | ||||
"""Very nasty problem with references held by multiple runs of a script. | ||||
|
r3917 | See: https://github.com/ipython/ipython/issues/141 | ||
|
r2414 | |||
In [1]: _ip.clear_main_mod_cache() | ||||
# random | ||||
In [2]: %run refbug | ||||
In [3]: call_f() | ||||
lowercased: hello | ||||
In [4]: %run refbug | ||||
In [5]: call_f() | ||||
lowercased: hello | ||||
lowercased: hello | ||||
""" | ||||
def doctest_run_builtins(): | ||||
r"""Check that %run doesn't damage __builtins__. | ||||
In [1]: import tempfile | ||||
In [2]: bid1 = id(__builtins__) | ||||
In [3]: fname = tempfile.mkstemp('.py')[1] | ||||
In [3]: f = open(fname,'w') | ||||
In [4]: f.write('pass\n') | ||||
In [5]: f.flush() | ||||
In [6]: t1 = type(__builtins__) | ||||
|
r2484 | In [7]: %run $fname | ||
|
r2414 | |||
In [7]: f.close() | ||||
In [8]: bid2 = id(__builtins__) | ||||
In [9]: t2 = type(__builtins__) | ||||
In [10]: t1 == t2 | ||||
Out[10]: True | ||||
In [10]: bid1 == bid2 | ||||
Out[10]: True | ||||
In [12]: try: | ||||
....: os.unlink(fname) | ||||
....: except: | ||||
....: pass | ||||
....: | ||||
""" | ||||
|
r3107 | |||
def doctest_reset_del(): | ||||
"""Test that resetting doesn't cause errors in __del__ methods. | ||||
In [2]: class A(object): | ||||
...: def __del__(self): | ||||
|
r3156 | ...: print str("Hi") | ||
|
r3107 | ...: | ||
In [3]: a = A() | ||||
In [4]: get_ipython().reset() | ||||
|
r3156 | Hi | ||
|
r3107 | |||
In [5]: 1+1 | ||||
Out[5]: 2 | ||||
""" | ||||
|
r2414 | |||
# For some tests, it will be handy to organize them in a class with a common | ||||
# setup that makes a temp file | ||||
|
r2415 | class TestMagicRunPass(tt.TempFileMixin): | ||
|
r2414 | |||
def setup(self): | ||||
"""Make a valid python temp file.""" | ||||
self.mktmp('pass\n') | ||||
def run_tmpfile(self): | ||||
_ip = get_ipython() | ||||
# This fails on Windows if self.tmpfile.name has spaces or "~" in it. | ||||
# See below and ticket https://bugs.launchpad.net/bugs/366353 | ||||
|
r2484 | _ip.magic('run %s' % self.fname) | ||
|
r2414 | |||
def test_builtins_id(self): | ||||
"""Check that %run doesn't damage __builtins__ """ | ||||
_ip = get_ipython() | ||||
# Test that the id of __builtins__ is not modified by %run | ||||
bid1 = id(_ip.user_ns['__builtins__']) | ||||
self.run_tmpfile() | ||||
bid2 = id(_ip.user_ns['__builtins__']) | ||||
tt.assert_equals(bid1, bid2) | ||||
def test_builtins_type(self): | ||||
"""Check that the type of __builtins__ doesn't change with %run. | ||||
However, the above could pass if __builtins__ was already modified to | ||||
be a dict (it should be a module) by a previous use of %run. So we | ||||
also check explicitly that it really is a module: | ||||
""" | ||||
_ip = get_ipython() | ||||
self.run_tmpfile() | ||||
tt.assert_equals(type(_ip.user_ns['__builtins__']),type(sys)) | ||||
def test_prompts(self): | ||||
"""Test that prompts correctly generate after %run""" | ||||
self.run_tmpfile() | ||||
_ip = get_ipython() | ||||
|
r2781 | p2 = str(_ip.displayhook.prompt2).strip() | ||
|
r2414 | nt.assert_equals(p2[:3], '...') | ||
|
r2415 | class TestMagicRunSimple(tt.TempFileMixin): | ||
|
r2414 | |||
def test_simpledef(self): | ||||
"""Test that simple class definitions work.""" | ||||
src = ("class foo: pass\n" | ||||
"def f(): return foo()") | ||||
self.mktmp(src) | ||||
|
r2451 | _ip.magic('run %s' % self.fname) | ||
|
r3752 | _ip.run_cell('t = isinstance(f(), foo)') | ||
|
r2414 | nt.assert_true(_ip.user_ns['t']) | ||
|
r2446 | |||
|
r2414 | def test_obj_del(self): | ||
"""Test that object's __del__ methods are called on exit.""" | ||||
src = ("class A(object):\n" | ||||
" def __del__(self):\n" | ||||
" print 'object A deleted'\n" | ||||
"a = A()\n") | ||||
self.mktmp(src) | ||||
tt.ipexec_validate(self.fname, 'object A deleted') | ||||
|
r3502 | |||
|
r3503 | @dec.skip_known_failure | ||
|
r3502 | def test_aggressive_namespace_cleanup(self): | ||
"""Test that namespace cleanup is not too aggressive GH-238 | ||||
|
r3503 | |||
Returning from another run magic deletes the namespace""" | ||||
|
r3502 | # see ticket https://github.com/ipython/ipython/issues/238 | ||
class secondtmp(tt.TempFileMixin): pass | ||||
empty = secondtmp() | ||||
empty.mktmp('') | ||||
src = ("ip = get_ipython()\n" | ||||
"for i in range(5):\n" | ||||
" try:\n" | ||||
" ip.magic('run %s')\n" | ||||
" except NameError, e:\n" | ||||
" print i;break\n" % empty.fname) | ||||
self.mktmp(src) | ||||
_ip.magic('run %s' % self.fname) | ||||
|
r3752 | _ip.run_cell('ip == get_ipython()') | ||
|
r3502 | tt.assert_equals(_ip.user_ns['i'], 5) | ||
|
r2414 | |||
|
r2446 | @dec.skip_win32 | ||
|
r2414 | def test_tclass(self): | ||
mydir = os.path.dirname(__file__) | ||||
tc = os.path.join(mydir, 'tclass') | ||||
src = ("%%run '%s' C-first\n" | ||||
|
r3763 | "%%run '%s' C-second\n" | ||
"%%run '%s' C-third\n") % (tc, tc, tc) | ||||
|
r2414 | self.mktmp(src, '.ipy') | ||
out = """\ | ||||
|
r3471 | ARGV 1-: [u'C-first'] | ||
ARGV 1-: [u'C-second'] | ||||
|
r2414 | tclass.py: deleting object: C-first | ||
|
r3763 | ARGV 1-: [u'C-third'] | ||
|
r3762 | tclass.py: deleting object: C-second | ||
|
r3763 | tclass.py: deleting object: C-third | ||
|
r2414 | """ | ||
tt.ipexec_validate(self.fname, out) | ||||