FM / tests

/opt/imh-python/lib/python3.9/site-packages/IPython/core/tests/

/opt/imh-python/lib/python3.9/site-packages/IPython/core/tests

Name	Size	Mode	Actions
daft_extension/	-	0755	rm
__pycache__/	-	0755	rm
2x2.jpg	331	0644	edit dl rm
2x2.png	71	0644	edit dl rm
bad_all.py	206	0644	edit dl rm
nonascii.py	135	0644	edit dl rm
nonascii2.py	155	0644	edit dl rm
print_argv.py	32	0644	edit dl rm
refbug.py	1494	0644	edit dl rm
simpleerr.py	583	0644	edit dl rm
tclass.py	921	0644	edit dl rm
test_alias.py	2007	0644	edit dl rm
test_application.py	2245	0644	edit dl rm
test_async_helpers.py	8647	0644	edit dl rm
test_autocall.py	1425	0644	edit dl rm
test_compilerop.py	2174	0644	edit dl rm
test_completer.py	60803	0644	edit dl rm
test_completerlib.py	6507	0644	edit dl rm
test_debugger.py	14501	0644	edit dl rm
test_display.py	15946	0644	edit dl rm
test_displayhook.py	3593	0644	edit dl rm
test_events.py	2287	0644	edit dl rm
test_exceptiongroup_tb.py	3599	0644	edit dl rm
test_extension.py	3026	0644	edit dl rm
test_formatters.py	14249	0644	edit dl rm
test_guarded_eval.py	15655	0644	edit dl rm
test_handlers.py	3286	0644	edit dl rm
test_history.py	11698	0644	edit dl rm
test_hooks.py	2316	0644	edit dl rm
test_imports.py	1153	0644	edit dl rm
test_inputsplitter.py	23213	0644	edit dl rm
test_inputtransformer.py	15755	0644	edit dl rm
test_inputtransformer2.py	11408	0644	edit dl rm
test_inputtransformer2_line.py	2971	0644	edit dl rm
test_interactiveshell.py	39660	0644	edit dl rm
test_iplib.py	6677	0644	edit dl rm
test_logger.py	776	0644	edit dl rm
test_magic.py	44359	0644	edit dl rm
test_magic_arguments.py	4778	0644	edit dl rm
test_magic_terminal.py	5996	0644	edit dl rm
test_oinspect.py	15674	0644	edit dl rm
test_page.py	772	0644	edit dl rm
test_paths.py	7081	0644	edit dl rm
test_prefilter.py	4214	0644	edit dl rm
test_profile.py	5064	0644	edit dl rm
test_prompts.py	845	0644	edit dl rm
test_pylabtools.py	7880	0644	edit dl rm
test_run.py	19369	0644	edit dl rm
test_shellapp.py	1892	0644	edit dl rm
test_splitinput.py	1227	0644	edit dl rm
test_ultratb.py	12026	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /opt/imh-python/lib/python3.9/site-packages/IPython/core/tests/test_run.py (19369B)

# encoding: utf-8
"""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.

Note that any test using `run -i` should make sure to do a `reset` afterwards,
as otherwise it may influence later tests.
"""

# Copyright (c) IPython Development Team.
# Distributed under the terms of the Modified BSD License.

import functools
import os
import platform
import random
import string
import sys
import textwrap
import unittest
from os.path import join as pjoin
from unittest.mock import patch

import pytest
from tempfile import TemporaryDirectory

from IPython.core import debugger
from IPython.testing import decorators as dec
from IPython.testing import tools as tt
from IPython.utils.io import capture_output

def doctest_refbug():
    """Very nasty problem with references held by multiple runs of a script.
    See: https://github.com/ipython/ipython/issues/141

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', encoding='utf-8')

In [4]: dummy= f.write('pass\n')

In [5]: f.flush()

In [6]: t1 = type(__builtins__)

In [7]: %run $fname

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
       ....:
    """

def doctest_run_option_parser():
    r"""Test option parser in %run.

In [1]: %run print_argv.py
    []

In [2]: %run print_argv.py print*.py
    ['print_argv.py']

In [3]: %run -G print_argv.py print*.py
    ['print*.py']

"""

@dec.skip_win32
def doctest_run_option_parser_for_posix():
    r"""Test option parser in %run (Linux/OSX specific).

You need double quote to escape glob in POSIX systems:

In [1]: %run print_argv.py print\\*.py
    ['print*.py']

You can't use quote to escape glob in POSIX systems:

In [2]: %run print_argv.py 'print*.py'
    ['print_argv.py']

"""

doctest_run_option_parser_for_posix.__skip_doctest__ = sys.platform == "win32"

@dec.skip_if_not_win32
def doctest_run_option_parser_for_windows():
    r"""Test option parser in %run (Windows specific).

In Windows, you can't escape ``*` `by backslash:

In [1]: %run print_argv.py print\\*.py
    ['print\\\\*.py']

You can use quote to escape glob:

In [2]: %run print_argv.py 'print*.py'
    ["'print*.py'"]

"""

doctest_run_option_parser_for_windows.__skip_doctest__ = sys.platform != "win32"

def doctest_reset_del():
    """Test that resetting doesn't cause errors in __del__ methods.

In [2]: class A(object):
       ...:     def __del__(self):
       ...:         print(str("Hi"))
       ...:

In [3]: a = A()

In [4]: get_ipython().reset(); import gc; x = gc.collect(0)
    Hi

In [5]: 1+1
    Out[5]: 2
    """

# For some tests, it will be handy to organize them in a class with a common
# setup that makes a temp file

class TestMagicRunPass(tt.TempFileMixin):

def setUp(self):
        content = "a = [1,2,3]\nb = 1"
        self.mktmp(content)

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
        _ip.run_line_magic("run", self.fname)

def run_tmpfile_p(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
        _ip.run_line_magic("run", "-p %s" % self.fname)

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__'])
        assert 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()
        assert type(_ip.user_ns["__builtins__"]) == type(sys)

def test_run_profile(self):
        """Test that the option -p, which invokes the profiler, do not
        crash by invoking execfile"""
        self.run_tmpfile_p()

def test_run_debug_twice(self):
        # https://github.com/ipython/ipython/issues/10028
        _ip = get_ipython()
        with tt.fake_input(["c"]):
            _ip.run_line_magic("run", "-d %s" % self.fname)
        with tt.fake_input(["c"]):
            _ip.run_line_magic("run", "-d %s" % self.fname)

def test_run_debug_twice_with_breakpoint(self):
        """Make a valid python temp file."""
        _ip = get_ipython()
        with tt.fake_input(["b 2", "c", "c"]):
            _ip.run_line_magic("run", "-d %s" % self.fname)

with tt.fake_input(["c"]):
            with tt.AssertNotPrints("KeyError"):
                _ip.run_line_magic("run", "-d %s" % self.fname)

class TestMagicRunSimple(tt.TempFileMixin):

def test_simpledef(self):
        """Test that simple class definitions work."""
        src = ("class foo: pass\n"
               "def f(): return foo()")
        self.mktmp(src)
        _ip.run_line_magic("run", str(self.fname))
        _ip.run_cell("t = isinstance(f(), foo)")
        assert _ip.user_ns["t"] is True

@pytest.mark.xfail(
        platform.python_implementation() == "PyPy",
        reason="expecting __del__ call on exit is unreliable and doesn't happen on PyPy",
    )
    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)
        err = None
        tt.ipexec_validate(self.fname, 'object A deleted', err)

def test_aggressive_namespace_cleanup(self):
        """Test that namespace cleanup is not too aggressive GH-238

Returning from another run magic deletes the namespace"""
        # see ticket https://github.com/ipython/ipython/issues/238

with tt.TempFileMixin() as empty:
            empty.mktmp("")
            # On Windows, the filename will have \users in it, so we need to use the
            # repr so that the \u becomes \\u.
            src = (
                "ip = get_ipython()\n"
                "for i in range(5):\n"
                "   try:\n"
                "       ip.magic(%r)\n"
                "   except NameError as e:\n"
                "       print(i)\n"
                "       break\n" % ("run " + empty.fname)
            )
            self.mktmp(src)
            _ip.run_line_magic("run", str(self.fname))
            _ip.run_cell("ip == get_ipython()")
            assert _ip.user_ns["i"] == 4

def test_run_second(self):
        """Test that running a second file doesn't clobber the first, gh-3547"""
        self.mktmp("avar = 1\n" "def afunc():\n" "  return avar\n")

with tt.TempFileMixin() as empty:
            empty.mktmp("")

_ip.run_line_magic("run", self.fname)
            _ip.run_line_magic("run", empty.fname)
            assert _ip.user_ns["afunc"]() == 1

def test_tclass(self):
        mydir = os.path.dirname(__file__)
        tc = os.path.join(mydir, "tclass")
        src = f"""\
import gc
%run "{tc}" C-first
gc.collect(0)
%run "{tc}" C-second
gc.collect(0)
%run "{tc}" C-third
gc.collect(0)
%reset -f
"""
        self.mktmp(src, ".ipy")
        out = """\
ARGV 1-: ['C-first']
ARGV 1-: ['C-second']
tclass.py: deleting object: C-first
ARGV 1-: ['C-third']
tclass.py: deleting object: C-second
tclass.py: deleting object: C-third
"""
        err = None
        tt.ipexec_validate(self.fname, out, err)

def test_run_i_after_reset(self):
        """Check that %run -i still works after %reset (gh-693)"""
        src = "yy = zz\n"
        self.mktmp(src)
        _ip.run_cell("zz = 23")
        try:
            _ip.run_line_magic("run", "-i %s" % self.fname)
            assert _ip.user_ns["yy"] == 23
        finally:
            _ip.run_line_magic("reset", "-f")

_ip.run_cell("zz = 23")
        try:
            _ip.run_line_magic("run", "-i %s" % self.fname)
            assert _ip.user_ns["yy"] == 23
        finally:
            _ip.run_line_magic("reset", "-f")

def test_unicode(self):
        """Check that files in odd encodings are accepted."""
        mydir = os.path.dirname(__file__)
        na = os.path.join(mydir, "nonascii.py")
        _ip.magic('run "%s"' % na)
        assert _ip.user_ns["u"] == "Ўт№Ф"

def test_run_py_file_attribute(self):
        """Test handling of `__file__` attribute in `%run <file>.py`."""
        src = "t = __file__\n"
        self.mktmp(src)
        _missing = object()
        file1 = _ip.user_ns.get("__file__", _missing)
        _ip.run_line_magic("run", self.fname)
        file2 = _ip.user_ns.get("__file__", _missing)

# Check that __file__ was equal to the filename in the script's
        # namespace.
        assert _ip.user_ns["t"] == self.fname

# Check that __file__ was not leaked back into user_ns.
        assert file1 == file2

def test_run_ipy_file_attribute(self):
        """Test handling of `__file__` attribute in `%run <file.ipy>`."""
        src = "t = __file__\n"
        self.mktmp(src, ext='.ipy')
        _missing = object()
        file1 = _ip.user_ns.get("__file__", _missing)
        _ip.run_line_magic("run", self.fname)
        file2 = _ip.user_ns.get("__file__", _missing)

# Check that __file__ was equal to the filename in the script's
        # namespace.
        assert _ip.user_ns["t"] == self.fname

# Check that __file__ was not leaked back into user_ns.
        assert file1 == file2

def test_run_formatting(self):
        """ Test that %run -t -N<N> does not raise a TypeError for N > 1."""
        src = "pass"
        self.mktmp(src)
        _ip.run_line_magic("run", "-t -N 1 %s" % self.fname)
        _ip.run_line_magic("run", "-t -N 10 %s" % self.fname)

def test_ignore_sys_exit(self):
        """Test the -e option to ignore sys.exit()"""
        src = "import sys; sys.exit(1)"
        self.mktmp(src)
        with tt.AssertPrints("SystemExit"):
            _ip.run_line_magic("run", self.fname)

with tt.AssertNotPrints("SystemExit"):
            _ip.run_line_magic("run", "-e %s" % self.fname)

def test_run_nb(self):
        """Test %run notebook.ipynb"""
        pytest.importorskip("nbformat")
        from nbformat import v4, writes
        nb = v4.new_notebook(
           cells=[
                v4.new_markdown_cell("The Ultimate Question of Everything"),
                v4.new_code_cell("answer=42")
            ]
        )
        src = writes(nb, version=4)
        self.mktmp(src, ext='.ipynb')

_ip.run_line_magic("run", self.fname)

assert _ip.user_ns["answer"] == 42

def test_run_nb_error(self):
        """Test %run notebook.ipynb error"""
        pytest.importorskip("nbformat")
        from nbformat import v4, writes

# %run when a file name isn't provided
        pytest.raises(Exception, _ip.magic, "run")

# %run when a file doesn't exist
        pytest.raises(Exception, _ip.magic, "run foobar.ipynb")

# %run on a notebook with an error
        nb = v4.new_notebook(
           cells=[
                v4.new_code_cell("0/0")
            ]
        )
        src = writes(nb, version=4)
        self.mktmp(src, ext='.ipynb')
        pytest.raises(Exception, _ip.magic, "run %s" % self.fname)

def test_file_options(self):
        src = ('import sys\n'
               'a = " ".join(sys.argv[1:])\n')
        self.mktmp(src)
        test_opts = "-x 3 --verbose"
        _ip.run_line_magic("run", "{0} {1}".format(self.fname, test_opts))
        assert _ip.user_ns["a"] == test_opts

class TestMagicRunWithPackage(unittest.TestCase):

def writefile(self, name, content):
        path = os.path.join(self.tempdir.name, name)
        d = os.path.dirname(path)
        if not os.path.isdir(d):
            os.makedirs(d)
        with open(path, "w", encoding="utf-8") as f:
            f.write(textwrap.dedent(content))

def setUp(self):
        self.package = package = 'tmp{0}'.format(''.join([random.choice(string.ascii_letters) for i in range(10)]))
        """Temporary  (probably) valid python package name."""

self.value = int(random.random() * 10000)

self.tempdir = TemporaryDirectory()
        self.__orig_cwd = os.getcwd()
        sys.path.insert(0, self.tempdir.name)

self.writefile(os.path.join(package, '__init__.py'), '')
        self.writefile(os.path.join(package, 'sub.py'), """
        x = {0!r}
        """.format(self.value))
        self.writefile(os.path.join(package, 'relative.py'), """
        from .sub import x
        """)
        self.writefile(os.path.join(package, 'absolute.py'), """
        from {0}.sub import x
        """.format(package))
        self.writefile(os.path.join(package, 'args.py'), """
        import sys
        a = " ".join(sys.argv[1:])
        """.format(package))

def tearDown(self):
        os.chdir(self.__orig_cwd)
        sys.path[:] = [p for p in sys.path if p != self.tempdir.name]
        self.tempdir.cleanup()

def check_run_submodule(self, submodule, opts=""):
        _ip.user_ns.pop("x", None)
        _ip.run_line_magic(
            "run", "{2} -m {0}.{1}".format(self.package, submodule, opts)
        )
        self.assertEqual(
            _ip.user_ns["x"],
            self.value,
            "Variable `x` is not loaded from module `{0}`.".format(submodule),
        )

def test_run_submodule_with_absolute_import(self):
        self.check_run_submodule('absolute')

def test_run_submodule_with_relative_import(self):
        """Run submodule that has a relative import statement (#2727)."""
        self.check_run_submodule('relative')

def test_prun_submodule_with_absolute_import(self):
        self.check_run_submodule('absolute', '-p')

def test_prun_submodule_with_relative_import(self):
        self.check_run_submodule('relative', '-p')

def with_fake_debugger(func):
        @functools.wraps(func)
        def wrapper(*args, **kwds):
            with patch.object(debugger.Pdb, 'run', staticmethod(eval)):
                return func(*args, **kwds)
        return wrapper

@with_fake_debugger
    def test_debug_run_submodule_with_absolute_import(self):
        self.check_run_submodule('absolute', '-d')

@with_fake_debugger
    def test_debug_run_submodule_with_relative_import(self):
        self.check_run_submodule('relative', '-d')

def test_module_options(self):
        _ip.user_ns.pop("a", None)
        test_opts = "-x abc -m test"
        _ip.run_line_magic("run", "-m {0}.args {1}".format(self.package, test_opts))
        assert _ip.user_ns["a"] == test_opts

def test_module_options_with_separator(self):
        _ip.user_ns.pop("a", None)
        test_opts = "-x abc -m test"
        _ip.run_line_magic("run", "-m {0}.args -- {1}".format(self.package, test_opts))
        assert _ip.user_ns["a"] == test_opts

def test_run__name__():
    with TemporaryDirectory() as td:
        path = pjoin(td, "foo.py")
        with open(path, "w", encoding="utf-8") as f:
            f.write("q = __name__")

_ip.user_ns.pop("q", None)
        _ip.run_line_magic("run", "{}".format(path))
        assert _ip.user_ns.pop("q") == "__main__"

_ip.run_line_magic("run", "-n {}".format(path))
        assert _ip.user_ns.pop("q") == "foo"

try:
            _ip.run_line_magic("run", "-i -n {}".format(path))
            assert _ip.user_ns.pop("q") == "foo"
        finally:
            _ip.run_line_magic("reset", "-f")

def test_run_tb():
    """Test traceback offset in %run"""
    with TemporaryDirectory() as td:
        path = pjoin(td, "foo.py")
        with open(path, "w", encoding="utf-8") as f:
            f.write(
                "\n".join(
                    [
                        "def foo():",
                        "    return bar()",
                        "def bar():",
                        "    raise RuntimeError('hello!')",
                        "foo()",
                    ]
                )
            )
        with capture_output() as io:
            _ip.run_line_magic("run", "{}".format(path))
        out = io.stdout
        assert "execfile" not in out
        assert "RuntimeError" in out
        assert out.count("---->") == 3
        del ip.user_ns['bar']
        del ip.user_ns['foo']

def test_multiprocessing_run():
    """Set we can run mutiprocesgin without messing up up main namespace

Note that import `nose.tools as nt` mdify the value s
    sys.module['__mp_main__'] so we need to temporarily set it to None to test
    the issue.
    """
    with TemporaryDirectory() as td:
        mpm = sys.modules.get('__mp_main__')
        sys.modules['__mp_main__'] = None
        try:
            path = pjoin(td, "test.py")
            with open(path, "w", encoding="utf-8") as f:
                f.write("import multiprocessing\nprint('hoy')")
            with capture_output() as io:
                _ip.run_line_magic('run', path)
                _ip.run_cell("i_m_undefined")
            out = io.stdout
            assert "hoy" in out
            assert "AttributeError" not in out
            assert "NameError" in out
            assert out.count("---->") == 1
        except:
            raise
        finally:
            sys.modules['__mp_main__'] = mpm

def test_script_tb():
    """Test traceback offset in `ipython script.py`"""
    with TemporaryDirectory() as td:
        path = pjoin(td, "foo.py")
        with open(path, "w", encoding="utf-8") as f:
            f.write(
                "\n".join(
                    [
                        "def foo():",
                        "    return bar()",
                        "def bar():",
                        "    raise RuntimeError('hello!')",
                        "foo()",
                    ]
                )
            )
        out, err = tt.ipexec(path)
        assert "execfile" not in out
        assert "RuntimeError" in out
        assert out.count("---->") == 3

cmd: