Tryag File Manager
Home
-
Turbo Force
Current Path :
/
usr
/
lib
/
python2.4
/
test
/
Upload File :
New :
File
Dir
//usr/lib/python2.4/test/test_pkg.py
# Test packages (dotted-name import) import sys, os, tempfile, traceback from os import mkdir, rmdir, extsep # Can't test if these fail del mkdir, rmdir from test.test_support import verify, verbose, TestFailed # Helpers to create and destroy hierarchies. def mkhier(root, descr): if not os.path.isdir(root): 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: mkdir(fullname) else: if verbose: print "write", fullname f = open(fullname, "w") f.write(contents) if contents and contents[-1] != '\n': f.write('\n') f.close() def mkdir(x): if verbose: print "mkdir", x os.mkdir(x) 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) rmdir(root) def rmdir(x): if verbose: print "rmdir", x os.rmdir(x) def fixdir(lst): try: lst.remove('__builtins__') except ValueError: pass return lst # Helper to run a test def runtest(hier, code): root = tempfile.mkdtemp() mkhier(root, hier) savepath = sys.path[:] fd, fname = tempfile.mkstemp(text=True) os.write(fd, code) os.close(fd) try: sys.path.insert(0, root) if verbose: print "sys.path =", sys.path try: execfile(fname, globals(), {}) except: traceback.print_exc(file=sys.stdout) finally: sys.path[:] = savepath os.unlink(fname) try: cleanout(root) except (os.error, IOError): pass # Test descriptions tests = [ ("t1", [("t1", None), ("t1 __init__"+os.extsep+"py", "")], "import t1"), ("t2", [ ("t2", None), ("t2 __init__"+os.extsep+"py", "'doc for t2'; print __name__, 'loading'"), ("t2 sub", None), ("t2 sub __init__"+os.extsep+"py", ""), ("t2 sub subsub", None), ("t2 sub subsub __init__"+os.extsep+"py", "print __name__, 'loading'; spam = 1"), ], """ import t2 print t2.__doc__ import t2.sub import t2.sub.subsub print t2.__name__, t2.sub.__name__, t2.sub.subsub.__name__ import t2 from t2 import * print dir() from t2 import sub from t2.sub import subsub from t2.sub.subsub import spam print sub.__name__, subsub.__name__ print sub.subsub.__name__ print dir() import t2.sub import t2.sub.subsub print t2.__name__, t2.sub.__name__, t2.sub.subsub.__name__ from t2 import * print dir() """), ("t3", [ ("t3", None), ("t3 __init__"+os.extsep+"py", "print __name__, 'loading'"), ("t3 sub", None), ("t3 sub __init__"+os.extsep+"py", ""), ("t3 sub subsub", None), ("t3 sub subsub __init__"+os.extsep+"py", "print __name__, 'loading'; spam = 1"), ], """ import t3.sub.subsub print t3.__name__, t3.sub.__name__, t3.sub.subsub.__name__ reload(t3) reload(t3.sub) reload(t3.sub.subsub) """), ("t4", [ ("t4"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (t4"+os.extsep+"py)'"), ("t4", None), ("t4 __init__"+os.extsep+"py", "print __name__, 'loading'"), ("t4 sub"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (sub"+os.extsep+"py)'"), ("t4 sub", None), ("t4 sub __init__"+os.extsep+"py", ""), ("t4 sub subsub"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (subsub"+os.extsep+"py)'"), ("t4 sub subsub", None), ("t4 sub subsub __init__"+os.extsep+"py", "print __name__, 'loading'; spam = 1"), ], """ from t4.sub.subsub import * print "t4.sub.subsub.spam =", spam """), ("t5", [ ("t5", None), ("t5 __init__"+os.extsep+"py", "import t5.foo"), ("t5 string"+os.extsep+"py", "print __name__, 'loading'; spam = 1"), ("t5 foo"+os.extsep+"py", "print __name__, 'loading'; import string; print string.spam"), ], """ import t5 from t5 import * print dir() import t5 print fixdir(dir(t5)) print fixdir(dir(t5.foo)) print fixdir(dir(t5.string)) """), ("t6", [ ("t6", None), ("t6 __init__"+os.extsep+"py", "__all__ = ['spam', 'ham', 'eggs']"), ("t6 spam"+os.extsep+"py", "print __name__, 'loading'"), ("t6 ham"+os.extsep+"py", "print __name__, 'loading'"), ("t6 eggs"+os.extsep+"py", "print __name__, 'loading'"), ], """ import t6 print fixdir(dir(t6)) from t6 import * print fixdir(dir(t6)) print dir() """), ("t7", [ ("t7"+os.extsep+"py", "print 'Importing t7"+os.extsep+"py'"), ("t7", None), ("t7 __init__"+os.extsep+"py", "print __name__, 'loading'"), ("t7 sub"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (sub"+os.extsep+"py)'"), ("t7 sub", None), ("t7 sub __init__"+os.extsep+"py", ""), ("t7 sub subsub"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (subsub"+os.extsep+"py)'"), ("t7 sub subsub", None), ("t7 sub subsub __init__"+os.extsep+"py", "print __name__, 'loading'; spam = 1"), ], """ t7, sub, subsub = None, None, None import t7 as tas print fixdir(dir(tas)) verify(not t7) from t7 import sub as subpar print fixdir(dir(subpar)) verify(not t7 and not sub) from t7.sub import subsub as subsubsub print fixdir(dir(subsubsub)) verify(not t7 and not sub and not subsub) from t7.sub.subsub import spam as ham print "t7.sub.subsub.spam =", ham verify(not t7 and not sub and not subsub) """), ] nontests = [ ("x5", [], ("import a" + ".a"*400)), ("x6", [], ("import a" + ".a"*499)), ("x7", [], ("import a" + ".a"*500)), ("x8", [], ("import a" + ".a"*1100)), ("x9", [], ("import " + "a"*400)), ("x10", [], ("import " + "a"*500)), ("x11", [], ("import " + "a"*998)), ("x12", [], ("import " + "a"*999)), ("x13", [], ("import " + "a"*999)), ("x14", [], ("import " + "a"*2000)), ] """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__) """ # Run the tests args = [] if __name__ == '__main__': args = sys.argv[1:] if args and args[0] == '-q': verbose = 0 del args[0] for name, hier, code in tests: if args and name not in args: print "skipping test", name continue print "running test", name runtest(hier, code) # Test import sys import imp try: import sys.imp except ImportError: # This is what we expect pass else: raise TestFailed, "No ImportError exception on 'import sys.imp'"