[pypy-svn] rev 644 - pypy/trunk/src/pypy/interpreter

[alex at codespeak.net]

Author: alex
Date: Thu May 29 11:59:34 2003
New Revision: 644

Added:
   pypy/trunk/src/pypy/interpreter/testsupport.py
Log:
needed here too, to let main.py switch objectspaces easily



Added: pypy/trunk/src/pypy/interpreter/testsupport.py
==============================================================================
--- (empty file)
+++ pypy/trunk/src/pypy/interpreter/testsupport.py	Thu May 29 11:59:34 2003
@@ -0,0 +1,63 @@
+"""
+Master version of testsupport.py: copy into any subdirectory of pypy
+from which scripts need to be run (typically all of the 'test' subdirs)
+so that any test can "import testsupport" to ensure the parent of pypy
+is on the sys.path -- so that "import pypy.etc.etc." always works.
+
+Also, this module exposes a member 'TestCase' that is unittest.TestCase
+or a subclass thereof supplying extra methods; and a function 'main'
+that is unittest.main or the equivalent.
+
+Furthermore, this module now exposes a member 'objspace' which is
+by default class pypy.objspace.trivial.TrivialObjSpace but can be
+set to use another objectspace instead; this allows tests to run
+under different objectspaces without needing to edit their sources.
+
+For this setting, use environment variable OBJSPACE and set it to
+a value such as 'pypy.objspace.trivial.TrivialObjSpace' (which is
+also the default if the environment variable is not found or empty
+or without any dot in it).
+
+When run as a script, runs all tests found in files called 'test_*.py'
+in the same directory.
+"""
+import sys, os
+
+try:
+    head = this_path = os.path.abspath(__file__)
+except NameError:
+    p = os.path.dirname(sys.argv[0])
+    if not p:
+        p = os.curdir
+    head = this_path = os.path.abspath(p)
+while 1:
+    head, tail = os.path.split(head)
+    if not tail:
+        raise EnvironmentError, "pypy not among parents of %r!" % this_path
+    elif tail.lower()=='pypy':
+        sys.path.insert(0, head)
+        break
+
+import pypy.interpreter.unittest_w
+TestCase = pypy.interpreter.unittest_w.TestCase_w
+import unittest
+main = unittest.main
+
+from pypy.interpreter import testtools
+
+objspace_path = os.environ.get('OBJSPACE')
+if not objspace_path or '.' not in objspace_path:
+    import pypy.objspace.trivial
+    objspace = pypy.objspace.trivial.TrivialObjSpace
+else:
+    objspace_pieces = objspace_path.split('.')
+    objspace_path = '.'.join(objspace_pieces[:-1])
+    objspace_module = __import__(objspace_path)
+    for piece in objspace_pieces[1:-1]:
+        objspace_module = getattr(objspace_module, piece)
+    objspace_classname = objspace_pieces[-1]
+    objspace = getattr(objspace_module, objspace_classname)
+
+if __name__ == '__main__':
+    runner = unittest.TextTestRunner()
+    runner.run(testtools.get_tests_for_dir(os.path.dirname(sys.argv[0])))