|
|
"""Tests for the key iplib module, where the main ipython class is defined.
|
|
|
"""
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Module imports
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
# stdlib
|
|
|
import os
|
|
|
import shutil
|
|
|
import tempfile
|
|
|
|
|
|
# third party
|
|
|
import nose.tools as nt
|
|
|
|
|
|
# our own packages
|
|
|
from IPython import ipapi, iplib
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Globals
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
# Useful global ipapi object and main IPython one. Unfortunately we have a
|
|
|
# long precedent of carrying the 'ipapi' global object which is injected into
|
|
|
# the system namespace as _ip, but that keeps a pointer to the actual IPython
|
|
|
# InteractiveShell instance, which is named IP. Since in testing we do need
|
|
|
# access to the real thing (we want to probe beyond what ipapi exposes), make
|
|
|
# here a global reference to each. In general, things that are exposed by the
|
|
|
# ipapi instance should be read from there, but we also will often need to use
|
|
|
# the actual IPython one.
|
|
|
|
|
|
# Get the public instance of IPython, and if it's None, make one so we can use
|
|
|
# it for testing
|
|
|
ip = ipapi.get()
|
|
|
if ip is None:
|
|
|
# IPython not running yet, make one from the testing machinery for
|
|
|
# consistency when the test suite is being run via iptest
|
|
|
from IPython.testing.plugin import ipdoctest
|
|
|
ip = ipapi.get()
|
|
|
|
|
|
IP = ip.IP # This is the actual IPython shell 'raw' object.
|
|
|
|
|
|
#-----------------------------------------------------------------------------
|
|
|
# Test functions
|
|
|
#-----------------------------------------------------------------------------
|
|
|
|
|
|
def test_reset():
|
|
|
"""reset must clear most namespaces."""
|
|
|
IP.reset() # first, it should run without error
|
|
|
# Then, check that most namespaces end up empty
|
|
|
for ns in IP.ns_refs_table:
|
|
|
if ns is IP.user_ns:
|
|
|
# The user namespace is reset with some data, so we can't check for
|
|
|
# it being empty
|
|
|
continue
|
|
|
nt.assert_equals(len(ns),0)
|
|
|
|
|
|
|
|
|
# make sure that user_setup can be run re-entrantly in 'install' mode.
|
|
|
def test_user_setup():
|
|
|
# use a lambda to pass kwargs to the generator
|
|
|
user_setup = lambda a,k: iplib.user_setup(*a,**k)
|
|
|
kw = dict(mode='install', interactive=False)
|
|
|
|
|
|
# Call the user setup and verify that the directory exists
|
|
|
yield user_setup, (ip.options.ipythondir,''), kw
|
|
|
yield os.path.isdir, ip.options.ipythondir
|
|
|
|
|
|
# Now repeat the operation with a non-existent directory. Check both that
|
|
|
# the call succeeds and that the directory is created.
|
|
|
tmpdir = tempfile.mktemp(prefix='ipython-test-')
|
|
|
# Use a try with an empty except because try/finally doesn't work with a
|
|
|
# yield in Python 2.4.
|
|
|
try:
|
|
|
yield user_setup, (tmpdir,''), kw
|
|
|
yield os.path.isdir, tmpdir
|
|
|
except:
|
|
|
pass
|
|
|
# Clean up the temp dir once done
|
|
|
shutil.rmtree(tmpdir)
|
|
|
|
