cpython-lru_cache-weakref: Lib/test/test

annotate Lib/test/test_runpy.py @ 78069:a8aa77458e12 3.2

Add c_size_t and c_ssize_t to the ctypes summary table.

author	Antoine Pitrou <solipsis@pitrou.net>
date	Thu, 12 Jul 2012 20:31:50 +0200
parents	cd1759711357
children	3b05cf877124 e240af1f0ae1

rev	line source
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	1 # Test the runpy module
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	2 import unittest
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	3 import os
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	4 import os.path
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	5 import sys
57315 eda56d0775ba Merged revisions 76362 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 57291 diff changeset	6 import re
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	7 import tempfile
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	8 import py_compile
62756 34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	9 from test.support import forget, make_legacy_pyc, run_unittest, unload, verbose
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	10 from test.script_helper import (
7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	11 make_pkg, make_script, make_zip_pkg, make_zip_script, temp_dir)
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	12
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	13
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	14 from runpy import _run_code, _run_module_code, run_module, run_path
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	15 # Note: This module can't safely test _run_module_as_main as it
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	16 # runs its tests in the current process, which would mess with the
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	17 # real __main__ module (usually test.regrtest)
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	18 # See test_cmd_line_script for a test that executes that code path
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	19
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	20 # Set up the test code and expected results
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	21
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	22 class RunModuleCodeTest(unittest.TestCase):
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	23 """Unit tests for runpy._run_code and runpy._run_module_code"""
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	24
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	25 expected_result = ["Top level assignment", "Lower level reference"]
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	26 test_source = (
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	27 "# Check basic code execution\n"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	28 "result = ['Top level assignment']\n"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	29 "def f():\n"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	30 " result.append('Lower level reference')\n"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	31 "f()\n"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	32 "# Check the sys module\n"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	33 "import sys\n"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	34 "run_argv0 = sys.argv[0]\n"
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	35 "run_name_in_sys_modules = __name__ in sys.modules\n"
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	36 "if run_name_in_sys_modules:\n"
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	37 " module_in_sys_modules = globals() is sys.modules[__name__].__dict__\n"
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	38 "# Check nested operation\n"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	39 "import runpy\n"
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	40 "nested = runpy._run_module_code('x=1\\n', mod_name='<run>')\n"
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	41 )
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	42
42954 27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	43 def test_run_code(self):
27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	44 saved_argv0 = sys.argv[0]
27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	45 d = _run_code(self.test_source, {})
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	46 self.assertEqual(d["result"], self.expected_result)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	47 self.assertIs(d["__name__"], None)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	48 self.assertIs(d["__file__"], None)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	49 self.assertIs(d["__cached__"], None)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	50 self.assertIs(d["__loader__"], None)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	51 self.assertIs(d["__package__"], None)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	52 self.assertIs(d["run_argv0"], saved_argv0)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	53 self.assertNotIn("run_name", d)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	54 self.assertIs(sys.argv[0], saved_argv0)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	55
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	56 def test_run_module_code(self):
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	57 initial = object()
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	58 name = "<Nonsense>"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	59 file = "Some other nonsense"
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	60 loader = "Now you're just being silly"
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	61 package = '' # Treat as a top level module
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	62 d1 = dict(initial=initial)
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	63 saved_argv0 = sys.argv[0]
42954 27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	64 d2 = _run_module_code(self.test_source,
27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	65 d1,
27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	66 name,
27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	67 file,
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	68 loader,
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	69 package)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	70 self.assertNotIn("result", d1)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	71 self.assertIs(d2["initial"], initial)
42954 27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	72 self.assertEqual(d2["result"], self.expected_result)
27f0289b6835 Merge the trunk changes in. Breaks socket.ssl for now. Thomas Wouters <thomas@python.org> parents: 42775 diff changeset	73 self.assertEqual(d2["nested"]["x"], 1)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	74 self.assertIs(d2["__name__"], name)
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 47201 diff changeset	75 self.assertTrue(d2["run_name_in_sys_modules"])
0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 47201 diff changeset	76 self.assertTrue(d2["module_in_sys_modules"])
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	77 self.assertIs(d2["__file__"], file)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	78 self.assertIs(d2["__cached__"], None)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	79 self.assertIs(d2["run_argv0"], file)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	80 self.assertIs(d2["__loader__"], loader)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	81 self.assertIs(d2["__package__"], package)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	82 self.assertIs(sys.argv[0], saved_argv0)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	83 self.assertNotIn(name, sys.modules)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	84
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	85
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	86 class RunModuleTest(unittest.TestCase):
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	87 """Unit tests for runpy.run_module"""
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	88
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	89 def expect_import_error(self, mod_name):
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	90 try:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	91 run_module(mod_name)
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	92 except ImportError:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	93 pass
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	94 else:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	95 self.fail("Expected import error for " + mod_name)
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	96
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	97 def test_invalid_names(self):
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	98 # Builtin module
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	99 self.expect_import_error("sys")
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	100 # Non-existent modules
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	101 self.expect_import_error("sys.imp.eric")
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	102 self.expect_import_error("os.path.half")
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	103 self.expect_import_error("a.bee")
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	104 self.expect_import_error(".howard")
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	105 self.expect_import_error("..eaten")
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	106 # Package without __main__.py
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	107 self.expect_import_error("multiprocessing")
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	108
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	109 def test_library_module(self):
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	110 run_module("runpy")
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	111
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	112 def _add_pkg_dir(self, pkg_dir):
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	113 os.mkdir(pkg_dir)
42622 4d2aa4bfb2d6 Remove RISCOS support Skip Montanaro <skip@pobox.com> parents: 42466 diff changeset	114 pkg_fname = os.path.join(pkg_dir, "__init__.py")
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	115 pkg_file = open(pkg_fname, "w")
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	116 pkg_file.close()
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	117 return pkg_fname
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	118
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	119 def _make_pkg(self, source, depth, mod_base="runpy_test"):
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	120 pkg_name = "__runpy_pkg__"
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	121 test_fname = mod_base+os.extsep+"py"
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	122 pkg_dir = sub_dir = tempfile.mkdtemp()
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	123 if verbose: print(" Package tree in:", sub_dir)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	124 sys.path.insert(0, pkg_dir)
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	125 if verbose: print(" Updated sys.path:", sys.path[0])
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	126 for i in range(depth):
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	127 sub_dir = os.path.join(sub_dir, pkg_name)
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	128 pkg_fname = self._add_pkg_dir(sub_dir)
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	129 if verbose: print(" Next level in:", sub_dir)
4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	130 if verbose: print(" Created:", pkg_fname)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	131 mod_fname = os.path.join(sub_dir, test_fname)
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	132 mod_file = open(mod_fname, "w")
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	133 mod_file.write(source)
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	134 mod_file.close()
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	135 if verbose: print(" Created:", mod_fname)
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	136 mod_name = (pkg_name+".")*depth + mod_base
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	137 return pkg_dir, mod_fname, mod_name
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	138
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	139 def _del_pkg(self, top, depth, mod_name):
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	140 for entry in list(sys.modules):
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	141 if entry.startswith("__runpy_pkg__"):
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	142 del sys.modules[entry]
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	143 if verbose: print(" Removed sys.modules entries")
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	144 del sys.path[0]
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	145 if verbose: print(" Removed sys.path entry")
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	146 for root, dirs, files in os.walk(top, topdown=False):
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	147 for name in files:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	148 try:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	149 os.remove(os.path.join(root, name))
40587 6b18a095c2e7 SF patch 1631942 by Collin Winter: Guido van Rossum <guido@python.org> parents: 37875 diff changeset	150 except OSError as ex:
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	151 if verbose: print(ex) # Persist with cleaning up
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	152 for name in dirs:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	153 fullname = os.path.join(root, name)
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	154 try:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	155 os.rmdir(fullname)
40587 6b18a095c2e7 SF patch 1631942 by Collin Winter: Guido van Rossum <guido@python.org> parents: 37875 diff changeset	156 except OSError as ex:
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	157 if verbose: print(ex) # Persist with cleaning up
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	158 try:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	159 os.rmdir(top)
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	160 if verbose: print(" Removed package tree")
40587 6b18a095c2e7 SF patch 1631942 by Collin Winter: Guido van Rossum <guido@python.org> parents: 37875 diff changeset	161 except OSError as ex:
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	162 if verbose: print(ex) # Persist with cleaning up
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	163
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	164 def _check_module(self, depth):
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	165 pkg_dir, mod_fname, mod_name = (
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	166 self._make_pkg("x=1\n", depth))
42775 d7517c306791 Merged revisions 57221-57391 via svnmerge from Guido van Rossum <guido@python.org> parents: 42622 diff changeset	167 forget(mod_name)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	168 try:
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	169 if verbose: print("Running from source:", mod_name)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	170 d1 = run_module(mod_name) # Read from source
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	171 self.assertIn("x", d1)
42188 81930eaae715 Patch by Christian Heimes to change self.assert_(x == y) into Guido van Rossum <guido@python.org> parents: 40775 diff changeset	172 self.assertEqual(d1["x"], 1)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	173 del d1 # Ensure __loader__ entry doesn't keep file open
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	174 __import__(mod_name)
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	175 os.remove(mod_fname)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	176 make_legacy_pyc(mod_fname)
62756 34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	177 unload(mod_name) # In case loader caches paths
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	178 if verbose: print("Running from compiled:", mod_name)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	179 d2 = run_module(mod_name) # Read from bytecode
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	180 self.assertIn("x", d2)
42188 81930eaae715 Patch by Christian Heimes to change self.assert_(x == y) into Guido van Rossum <guido@python.org> parents: 40775 diff changeset	181 self.assertEqual(d2["x"], 1)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	182 del d2 # Ensure __loader__ entry doesn't keep file open
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	183 finally:
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	184 self._del_pkg(pkg_dir, depth, mod_name)
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	185 if verbose: print("Module executed successfully")
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	186
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	187 def _check_package(self, depth):
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	188 pkg_dir, mod_fname, mod_name = (
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	189 self._make_pkg("x=1\n", depth, "__main__"))
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	190 pkg_name, _, _ = mod_name.rpartition(".")
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	191 forget(mod_name)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	192 try:
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	193 if verbose: print("Running from source:", pkg_name)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	194 d1 = run_module(pkg_name) # Read from source
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	195 self.assertIn("x", d1)
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	196 self.assertTrue(d1["x"] == 1)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	197 del d1 # Ensure __loader__ entry doesn't keep file open
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	198 __import__(mod_name)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	199 os.remove(mod_fname)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	200 make_legacy_pyc(mod_fname)
62756 34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	201 unload(mod_name) # In case loader caches paths
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	202 if verbose: print("Running from compiled:", pkg_name)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	203 d2 = run_module(pkg_name) # Read from bytecode
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	204 self.assertIn("x", d2)
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	205 self.assertTrue(d2["x"] == 1)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	206 del d2 # Ensure __loader__ entry doesn't keep file open
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	207 finally:
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	208 self._del_pkg(pkg_dir, depth, pkg_name)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	209 if verbose: print("Package executed successfully")
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	210
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	211 def _add_relative_modules(self, base_dir, source, depth):
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	212 if depth <= 1:
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	213 raise ValueError("Relative module test needs depth > 1")
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	214 pkg_name = "__runpy_pkg__"
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	215 module_dir = base_dir
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	216 for i in range(depth):
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	217 parent_dir = module_dir
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	218 module_dir = os.path.join(module_dir, pkg_name)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	219 # Add sibling module
42622 4d2aa4bfb2d6 Remove RISCOS support Skip Montanaro <skip@pobox.com> parents: 42466 diff changeset	220 sibling_fname = os.path.join(module_dir, "sibling.py")
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	221 sibling_file = open(sibling_fname, "w")
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	222 sibling_file.close()
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	223 if verbose: print(" Added sibling module:", sibling_fname)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	224 # Add nephew module
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	225 uncle_dir = os.path.join(parent_dir, "uncle")
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	226 self._add_pkg_dir(uncle_dir)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	227 if verbose: print(" Added uncle package:", uncle_dir)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	228 cousin_dir = os.path.join(uncle_dir, "cousin")
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	229 self._add_pkg_dir(cousin_dir)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	230 if verbose: print(" Added cousin package:", cousin_dir)
42622 4d2aa4bfb2d6 Remove RISCOS support Skip Montanaro <skip@pobox.com> parents: 42466 diff changeset	231 nephew_fname = os.path.join(cousin_dir, "nephew.py")
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	232 nephew_file = open(nephew_fname, "w")
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	233 nephew_file.close()
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	234 if verbose: print(" Added nephew module:", nephew_fname)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	235
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	236 def _check_relative_imports(self, depth, run_name=None):
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	237 contents = r"""\
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	238 from __future__ import absolute_import
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	239 from . import sibling
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	240 from ..uncle.cousin import nephew
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	241 """
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	242 pkg_dir, mod_fname, mod_name = (
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	243 self._make_pkg(contents, depth))
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	244 try:
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	245 self._add_relative_modules(pkg_dir, contents, depth)
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	246 pkg_name = mod_name.rpartition('.')[0]
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	247 if verbose: print("Running from source:", mod_name)
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	248 d1 = run_module(mod_name, run_name=run_name) # Read from source
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	249 self.assertIn("__package__", d1)
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 47201 diff changeset	250 self.assertTrue(d1["__package__"] == pkg_name)
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	251 self.assertIn("sibling", d1)
f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	252 self.assertIn("nephew", d1)
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	253 del d1 # Ensure __loader__ entry doesn't keep file open
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	254 __import__(mod_name)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	255 os.remove(mod_fname)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	256 make_legacy_pyc(mod_fname)
62756 34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	257 unload(mod_name) # In case the loader caches paths
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	258 if verbose: print("Running from compiled:", mod_name)
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	259 d2 = run_module(mod_name, run_name=run_name) # Read from bytecode
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	260 self.assertIn("__package__", d2)
55593 0bb232078b8c Merged revisions 73715 via svnmerge from Georg Brandl <georg@python.org> parents: 47201 diff changeset	261 self.assertTrue(d2["__package__"] == pkg_name)
58415 f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	262 self.assertIn("sibling", d2)
f6a626f34840 use assert[Not]In where appropriate Benjamin Peterson <benjamin@python.org> parents: 57450 diff changeset	263 self.assertIn("nephew", d2)
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	264 del d2 # Ensure __loader__ entry doesn't keep file open
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	265 finally:
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	266 self._del_pkg(pkg_dir, depth, mod_name)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	267 if verbose: print("Module executed successfully")
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	268
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	269 def test_run_module(self):
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	270 for depth in range(4):
40775 4cdf21a0dbbc Fix most trivially-findable print statements. Guido van Rossum <guido@python.org> parents: 40587 diff changeset	271 if verbose: print("Testing package depth:", depth)
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	272 self._check_module(depth)
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	273
57450 0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	274 def test_run_package(self):
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	275 for depth in range(1, 4):
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	276 if verbose: print("Testing package depth:", depth)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	277 self._check_package(depth)
0db67153d25f Merged revisions 75264,75268,75293,75318,75391-75392,75436,75478,75971,76003,76058,76140-76141,76231,76380,76428-76429 via svnmerge from Benjamin Peterson <benjamin@python.org> parents: 57315 diff changeset	278
42466 3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	279 def test_explicit_relative_import(self):
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	280 for depth in range(2, 5):
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	281 if verbose: print("Testing relative imports at depth:", depth)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	282 self._check_relative_imports(depth)
3fc3cedc49ca Merged revisions 56753-56781 via svnmerge from Guido van Rossum <guido@python.org> parents: 42188 diff changeset	283
44129 b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	284 def test_main_relative_import(self):
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	285 for depth in range(2, 5):
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	286 if verbose: print("Testing main relative imports at depth:", depth)
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	287 self._check_relative_imports(depth, "__main__")
b637349336fd Merged revisions 59275-59303 via svnmerge from Christian Heimes <christian@cheimes.de> parents: 42954 diff changeset	288
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	289
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	290 class RunPathTest(unittest.TestCase):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	291 """Unit tests for runpy.run_path"""
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	292 # Based on corresponding tests in test_cmd_line_script
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	293
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	294 test_source = """\
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	295 # Script may be run with optimisation enabled, so don't rely on assert
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	296 # statements being executed
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	297 def assertEqual(lhs, rhs):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	298 if lhs != rhs:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	299 raise AssertionError('%r != %r' % (lhs, rhs))
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	300 def assertIs(lhs, rhs):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	301 if lhs is not rhs:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	302 raise AssertionError('%r is not %r' % (lhs, rhs))
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	303 # Check basic code execution
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	304 result = ['Top level assignment']
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	305 def f():
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	306 result.append('Lower level reference')
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	307 f()
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	308 assertEqual(result, ['Top level assignment', 'Lower level reference'])
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	309 # Check the sys module
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	310 import sys
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	311 assertIs(globals(), sys.modules[__name__].__dict__)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	312 argv0 = sys.argv[0]
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	313 """
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	314
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	315 def _make_test_script(self, script_dir, script_basename, source=None):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	316 if source is None:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	317 source = self.test_source
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	318 return make_script(script_dir, script_basename, source)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	319
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	320 def _check_script(self, script_name, expected_name, expected_file,
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	321 expected_argv0, expected_package):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	322 result = run_path(script_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	323 self.assertEqual(result["__name__"], expected_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	324 self.assertEqual(result["__file__"], expected_file)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	325 self.assertEqual(result["__cached__"], None)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	326 self.assertIn("argv0", result)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	327 self.assertEqual(result["argv0"], expected_argv0)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	328 self.assertEqual(result["__package__"], expected_package)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	329
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	330 def _check_import_error(self, script_name, msg):
57315 eda56d0775ba Merged revisions 76362 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 57291 diff changeset	331 msg = re.escape(msg)
66581 38bec90fd8ab #10273: Rename assertRegexpMatches and assertRaisesRegexp to assertRegex and assertRaisesRegex. Ezio Melotti <ezio.melotti@gmail.com> parents: 62756 diff changeset	332 self.assertRaisesRegex(ImportError, msg, run_path, script_name)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	333
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	334 def test_basic_script(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	335 with temp_dir() as script_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	336 mod_name = 'script'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	337 script_name = self._make_test_script(script_dir, mod_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	338 self._check_script(script_name, "<run_path>", script_name,
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	339 script_name, None)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	340
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	341 def test_script_compiled(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	342 with temp_dir() as script_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	343 mod_name = 'script'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	344 script_name = self._make_test_script(script_dir, mod_name)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	345 compiled_name = py_compile.compile(script_name, doraise=True)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	346 os.remove(script_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	347 self._check_script(compiled_name, "<run_path>", compiled_name,
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	348 compiled_name, None)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	349
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	350 def test_directory(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	351 with temp_dir() as script_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	352 mod_name = '__main__'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	353 script_name = self._make_test_script(script_dir, mod_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	354 self._check_script(script_dir, "<run_path>", script_name,
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	355 script_dir, '')
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	356
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	357 def test_directory_compiled(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	358 with temp_dir() as script_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	359 mod_name = '__main__'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	360 script_name = self._make_test_script(script_dir, mod_name)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	361 compiled_name = py_compile.compile(script_name, doraise=True)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	362 os.remove(script_name)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	363 legacy_pyc = make_legacy_pyc(script_name)
7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	364 self._check_script(script_dir, "<run_path>", legacy_pyc,
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	365 script_dir, '')
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	366
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	367 def test_directory_error(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	368 with temp_dir() as script_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	369 mod_name = 'not_main'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	370 script_name = self._make_test_script(script_dir, mod_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	371 msg = "can't find '__main__' module in %r" % script_dir
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	372 self._check_import_error(script_dir, msg)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	373
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	374 def test_zipfile(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	375 with temp_dir() as script_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	376 mod_name = '__main__'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	377 script_name = self._make_test_script(script_dir, mod_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	378 zip_name, fname = make_zip_script(script_dir, 'test_zip', script_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	379 self._check_script(zip_name, "<run_path>", fname, zip_name, '')
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	380
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	381 def test_zipfile_compiled(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	382 with temp_dir() as script_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	383 mod_name = '__main__'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	384 script_name = self._make_test_script(script_dir, mod_name)
60648 7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	385 compiled_name = py_compile.compile(script_name, doraise=True)
7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	386 zip_name, fname = make_zip_script(script_dir, 'test_zip',
7b69e630d237 PEP 3147 Barry Warsaw <barry@python.org> parents: 58415 diff changeset	387 compiled_name)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	388 self._check_script(zip_name, "<run_path>", fname, zip_name, '')
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	389
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	390 def test_zipfile_error(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	391 with temp_dir() as script_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	392 mod_name = 'not_main'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	393 script_name = self._make_test_script(script_dir, mod_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	394 zip_name, fname = make_zip_script(script_dir, 'test_zip', script_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	395 msg = "can't find '__main__' module in %r" % zip_name
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	396 self._check_import_error(zip_name, msg)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	397
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	398 def test_main_recursion_error(self):
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	399 with temp_dir() as script_dir, temp_dir() as dummy_dir:
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	400 mod_name = '__main__'
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	401 source = ("import runpy\n"
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	402 "runpy.run_path(%r)\n") % dummy_dir
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	403 script_name = self._make_test_script(script_dir, mod_name, source)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	404 zip_name, fname = make_zip_script(script_dir, 'test_zip', script_name)
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	405 msg = "recursion depth exceeded"
66581 38bec90fd8ab #10273: Rename assertRegexpMatches and assertRaisesRegexp to assertRegex and assertRaisesRegex. Ezio Melotti <ezio.melotti@gmail.com> parents: 62756 diff changeset	406 self.assertRaisesRegex(RuntimeError, msg, run_path, zip_name)
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	407
71164 cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	408 def test_encoding(self):
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	409 with temp_dir() as script_dir:
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	410 filename = os.path.join(script_dir, 'script.py')
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	411 with open(filename, 'w', encoding='latin1') as f:
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	412 f.write("""
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	413 #coding:latin1
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	414 "non-ASCII: h\xe9"
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	415 """)
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	416 result = run_path(filename)
cd1759711357 Issue #12451: runpy: run_path() now opens the Python script in binary mode, Victor Stinner <victor.stinner@haypocalc.com> parents: 66581 diff changeset	417 self.assertEqual(result['__doc__'], "non-ASCII: h\xe9")
57291 fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	418
fd417cd64341 Merged revisions 76286-76287,76289-76294,76296-76299,76301-76305,76307,76310-76311,76313-76322 via svnmerge from Nick Coghlan <ncoghlan@gmail.com> parents: 55593 diff changeset	419
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	420 def test_main():
62756 34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	421 run_unittest(
34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	422 RunModuleCodeTest,
34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	423 RunModuleTest,
34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	424 RunPathTest
34d827249ce0 Make importlib.abc.SourceLoader the primary mechanism for importlib. Brett Cannon <bcannon@gmail.com> parents: 60648 diff changeset	425 )
37875 a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	426
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	427 if __name__ == "__main__":
a4c70f3968be Merge part of the trunk changes into the p3yk branch. This merges from 43030 Thomas Wouters <thomas@python.org> parents: diff changeset	428 test_main()

Mercurial > cpython-lru_cache-weakref

annotate Lib/test/test_runpy.py @ 78069:a8aa77458e12 3.2