mirror of
https://github.com/python/cpython.git
synced 2024-11-26 03:14:27 +08:00
378c0cf5ab
svn+ssh://pythondev@svn.python.org/python/trunk ........ r78351 | r.david.murray | 2010-02-22 19:24:49 -0500 (Mon, 22 Feb 2010) | 5 lines Issue 6292: for the moment at least, the test suite passes if run with -OO. Tests requiring docstrings are skipped. Patch by Brian Curtin, thanks to Matias Torchinsky for helping review and improve the patch. ........
295 lines
9.1 KiB
Python
295 lines
9.1 KiB
Python
# Test packages (dotted-name import)
|
|
|
|
import sys
|
|
import os
|
|
import tempfile
|
|
import textwrap
|
|
import unittest
|
|
from test import support
|
|
|
|
|
|
# Helpers to create and destroy hierarchies.
|
|
|
|
def cleanout(root):
|
|
names = os.listdir(root)
|
|
for name in names:
|
|
fullname = os.path.join(root, name)
|
|
if os.path.isdir(fullname) and not os.path.islink(fullname):
|
|
cleanout(fullname)
|
|
else:
|
|
os.remove(fullname)
|
|
os.rmdir(root)
|
|
|
|
def fixdir(lst):
|
|
if "__builtins__" in lst:
|
|
lst.remove("__builtins__")
|
|
return lst
|
|
|
|
|
|
# XXX Things to test
|
|
#
|
|
# import package without __init__
|
|
# import package with __init__
|
|
# __init__ importing submodule
|
|
# __init__ importing global module
|
|
# __init__ defining variables
|
|
# submodule importing other submodule
|
|
# submodule importing global module
|
|
# submodule import submodule via global name
|
|
# from package import submodule
|
|
# from package import subpackage
|
|
# from package import variable (defined in __init__)
|
|
# from package import * (defined in __init__)
|
|
|
|
|
|
class TestPkg(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.root = None
|
|
self.pkgname = None
|
|
self.syspath = list(sys.path)
|
|
self.modules_before = support.modules_setup()
|
|
|
|
def tearDown(self):
|
|
sys.path[:] = self.syspath
|
|
support.modules_cleanup(*self.modules_before)
|
|
if self.root: # Only clean if the test was actually run
|
|
cleanout(self.root)
|
|
|
|
# delete all modules concerning the tested hiearchy
|
|
if self.pkgname:
|
|
modules = [name for name in sys.modules
|
|
if self.pkgname in name.split('.')]
|
|
for name in modules:
|
|
del sys.modules[name]
|
|
|
|
def run_code(self, code):
|
|
exec(textwrap.dedent(code), globals(), {"self": self})
|
|
|
|
def mkhier(self, descr):
|
|
root = tempfile.mkdtemp()
|
|
sys.path.insert(0, root)
|
|
if not os.path.isdir(root):
|
|
os.mkdir(root)
|
|
for name, contents in descr:
|
|
comps = name.split()
|
|
fullname = root
|
|
for c in comps:
|
|
fullname = os.path.join(fullname, c)
|
|
if contents is None:
|
|
os.mkdir(fullname)
|
|
else:
|
|
f = open(fullname, "w")
|
|
f.write(contents)
|
|
if contents and contents[-1] != '\n':
|
|
f.write('\n')
|
|
f.close()
|
|
self.root = root
|
|
# package name is the name of the first item
|
|
self.pkgname = descr[0][0]
|
|
|
|
def test_1(self):
|
|
hier = [("t1", None), ("t1 __init__.py", "")]
|
|
self.mkhier(hier)
|
|
import t1
|
|
|
|
def test_2(self):
|
|
hier = [
|
|
("t2", None),
|
|
("t2 __init__.py", "'doc for t2'"),
|
|
("t2 sub", None),
|
|
("t2 sub __init__.py", ""),
|
|
("t2 sub subsub", None),
|
|
("t2 sub subsub __init__.py", "spam = 1"),
|
|
]
|
|
self.mkhier(hier)
|
|
|
|
import t2.sub
|
|
import t2.sub.subsub
|
|
self.assertEqual(t2.__name__, "t2")
|
|
self.assertEqual(t2.sub.__name__, "t2.sub")
|
|
self.assertEqual(t2.sub.subsub.__name__, "t2.sub.subsub")
|
|
|
|
# This exec crap is needed because Py3k forbids 'import *' outside
|
|
# of module-scope and __import__() is insufficient for what we need.
|
|
s = """
|
|
import t2
|
|
from t2 import *
|
|
self.assertEqual(dir(), ['self', 'sub', 't2'])
|
|
"""
|
|
self.run_code(s)
|
|
|
|
from t2 import sub
|
|
from t2.sub import subsub
|
|
from t2.sub.subsub import spam
|
|
self.assertEqual(sub.__name__, "t2.sub")
|
|
self.assertEqual(subsub.__name__, "t2.sub.subsub")
|
|
self.assertEqual(sub.subsub.__name__, "t2.sub.subsub")
|
|
for name in ['spam', 'sub', 'subsub', 't2']:
|
|
self.assertTrue(locals()["name"], "Failed to import %s" % name)
|
|
|
|
import t2.sub
|
|
import t2.sub.subsub
|
|
self.assertEqual(t2.__name__, "t2")
|
|
self.assertEqual(t2.sub.__name__, "t2.sub")
|
|
self.assertEqual(t2.sub.subsub.__name__, "t2.sub.subsub")
|
|
|
|
s = """
|
|
from t2 import *
|
|
self.assertTrue(dir(), ['self', 'sub'])
|
|
"""
|
|
self.run_code(s)
|
|
|
|
def test_3(self):
|
|
hier = [
|
|
("t3", None),
|
|
("t3 __init__.py", ""),
|
|
("t3 sub", None),
|
|
("t3 sub __init__.py", ""),
|
|
("t3 sub subsub", None),
|
|
("t3 sub subsub __init__.py", "spam = 1"),
|
|
]
|
|
self.mkhier(hier)
|
|
|
|
import t3.sub.subsub
|
|
self.assertEqual(t3.__name__, "t3")
|
|
self.assertEqual(t3.sub.__name__, "t3.sub")
|
|
self.assertEqual(t3.sub.subsub.__name__, "t3.sub.subsub")
|
|
|
|
def test_4(self):
|
|
hier = [
|
|
("t4.py", "raise RuntimeError('Shouldnt load t4.py')"),
|
|
("t4", None),
|
|
("t4 __init__.py", ""),
|
|
("t4 sub.py", "raise RuntimeError('Shouldnt load sub.py')"),
|
|
("t4 sub", None),
|
|
("t4 sub __init__.py", ""),
|
|
("t4 sub subsub.py",
|
|
"raise RuntimeError('Shouldnt load subsub.py')"),
|
|
("t4 sub subsub", None),
|
|
("t4 sub subsub __init__.py", "spam = 1"),
|
|
]
|
|
self.mkhier(hier)
|
|
|
|
s = """
|
|
from t4.sub.subsub import *
|
|
self.assertEqual(spam, 1)
|
|
"""
|
|
self.run_code(s)
|
|
|
|
def test_5(self):
|
|
hier = [
|
|
("t5", None),
|
|
("t5 __init__.py", "import t5.foo"),
|
|
("t5 string.py", "spam = 1"),
|
|
("t5 foo.py",
|
|
"from . import string; assert string.spam == 1"),
|
|
]
|
|
self.mkhier(hier)
|
|
|
|
import t5
|
|
s = """
|
|
from t5 import *
|
|
self.assertEqual(dir(), ['foo', 'self', 'string', 't5'])
|
|
"""
|
|
self.run_code(s)
|
|
|
|
import t5
|
|
self.assertEqual(fixdir(dir(t5)),
|
|
['__doc__', '__file__', '__name__',
|
|
'__package__', '__path__', 'foo', 'string', 't5'])
|
|
self.assertEqual(fixdir(dir(t5.foo)),
|
|
['__doc__', '__file__', '__name__', '__package__',
|
|
'string'])
|
|
self.assertEqual(fixdir(dir(t5.string)),
|
|
['__doc__', '__file__', '__name__','__package__',
|
|
'spam'])
|
|
|
|
def test_6(self):
|
|
hier = [
|
|
("t6", None),
|
|
("t6 __init__.py",
|
|
"__all__ = ['spam', 'ham', 'eggs']"),
|
|
("t6 spam.py", ""),
|
|
("t6 ham.py", ""),
|
|
("t6 eggs.py", ""),
|
|
]
|
|
self.mkhier(hier)
|
|
|
|
import t6
|
|
self.assertEqual(fixdir(dir(t6)),
|
|
['__all__', '__doc__', '__file__',
|
|
'__name__', '__package__', '__path__'])
|
|
s = """
|
|
import t6
|
|
from t6 import *
|
|
self.assertEqual(fixdir(dir(t6)),
|
|
['__all__', '__doc__', '__file__',
|
|
'__name__', '__package__', '__path__',
|
|
'eggs', 'ham', 'spam'])
|
|
self.assertEqual(dir(), ['eggs', 'ham', 'self', 'spam', 't6'])
|
|
"""
|
|
self.run_code(s)
|
|
|
|
def test_7(self):
|
|
hier = [
|
|
("t7.py", ""),
|
|
("t7", None),
|
|
("t7 __init__.py", ""),
|
|
("t7 sub.py",
|
|
"raise RuntimeError('Shouldnt load sub.py')"),
|
|
("t7 sub", None),
|
|
("t7 sub __init__.py", ""),
|
|
("t7 sub .py",
|
|
"raise RuntimeError('Shouldnt load subsub.py')"),
|
|
("t7 sub subsub", None),
|
|
("t7 sub subsub __init__.py",
|
|
"spam = 1"),
|
|
]
|
|
self.mkhier(hier)
|
|
|
|
|
|
t7, sub, subsub = None, None, None
|
|
import t7 as tas
|
|
self.assertEqual(fixdir(dir(tas)),
|
|
['__doc__', '__file__', '__name__',
|
|
'__package__', '__path__'])
|
|
self.assertFalse(t7)
|
|
from t7 import sub as subpar
|
|
self.assertEqual(fixdir(dir(subpar)),
|
|
['__doc__', '__file__', '__name__',
|
|
'__package__', '__path__'])
|
|
self.assertFalse(t7)
|
|
self.assertFalse(sub)
|
|
from t7.sub import subsub as subsubsub
|
|
self.assertEqual(fixdir(dir(subsubsub)),
|
|
['__doc__', '__file__', '__name__',
|
|
'__package__', '__path__', 'spam'])
|
|
self.assertFalse(t7)
|
|
self.assertFalse(sub)
|
|
self.assertFalse(subsub)
|
|
from t7.sub.subsub import spam as ham
|
|
self.assertEqual(ham, 1)
|
|
self.assertFalse(t7)
|
|
self.assertFalse(sub)
|
|
self.assertFalse(subsub)
|
|
|
|
@unittest.skipIf(sys.flags.optimize >= 2,
|
|
"Docstrings are omitted with -O2 and above")
|
|
def test_8(self):
|
|
hier = [
|
|
("t8", None),
|
|
("t8 __init__"+os.extsep+"py", "'doc for t8'"),
|
|
]
|
|
self.mkhier(hier)
|
|
|
|
import t8
|
|
self.assertEqual(t8.__doc__, "doc for t8")
|
|
|
|
def test_main():
|
|
support.run_unittest(__name__)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|