Renamed scipy.test module to scipy.testing. Clean up testing.

1 files changed, 261 insertions, 0 deletions

diff --git a/scipy/testing/testing.py b/scipy/testing/testing.py
new file mode 100644
index 000000000..4a2dfad9c
--- /dev/null
+++ b/scipy/testing/testing.py
@@ -0,0 +1,261 @@
+"""
+
+
+
+
+
+
+"""
+
+__all__ = []
+
+import os,sys,time,glob,string,traceback,unittest
+import types
+import imp
+
+#
+# Imports from scipy.base must be done at the end of this file.
+#
+
+DEBUG = 0
+
+
+__all__.append('ScipyTestCase')
+
+
+
+
+
+
+#------------
+
+
+
+#------------
+        
+def remove_ignored_patterns(files,pattern):
+    from fnmatch import fnmatch
+    good_files = []
+    for file in files:
+        if not fnmatch(file,pattern):
+            good_files.append(file)
+    return good_files
+
+def remove_ignored_files(original,ignored_files,cur_dir):
+    """ This is actually expanded to do pattern matching.
+
+    """
+    if not ignored_files: ignored_files = []
+    ignored_modules = map(lambda x: x+'.py',ignored_files)
+    ignored_packages = ignored_files[:]
+    # always ignore setup.py and __init__.py files
+    ignored_files = ['setup.py','setup_*.py','__init__.py']
+    ignored_files += ignored_modules + ignored_packages
+    ignored_files = map(lambda x,cur_dir=cur_dir: os.path.join(cur_dir,x),
+                        ignored_files)
+    #print 'ignored:', ignored_files
+    #good_files = filter(lambda x,ignored = ignored_files: x not in ignored,
+    #                    original)
+    good_files = original
+    for pattern in ignored_files:
+        good_files = remove_ignored_patterns(good_files,pattern)
+
+    return good_files
+
+__all__.append('harvest_modules')
+def harvest_modules(package,ignore=None):
+    """* Retreive a list of all modules that live within a package.
+
+         Only retreive files that are immediate children of the
+         package -- do not recurse through child packages or
+         directories.  The returned list contains actual modules, not
+         just their names.
+    *"""
+    d,f = os.path.split(package.__file__)
+
+    # go through the directory and import every py file there.
+    common_dir = os.path.join(d,'*.py')
+    py_files = glob.glob(common_dir)
+    #py_files.remove(os.path.join(d,'__init__.py'))
+    #py_files.remove(os.path.join(d,'setup.py'))
+
+    py_files = remove_ignored_files(py_files,ignore,d)
+    #print 'py_files:', py_files
+    try:
+        prefix = package.__name__
+    except:
+        prefix = ''
+
+    all_modules = []
+    for file in py_files:
+        d,f = os.path.split(file)
+        base,ext =  os.path.splitext(f)
+        mod = prefix + '.' + base
+        #print 'module: import ' + mod
+        try:
+            exec ('import ' + mod)
+            all_modules.append(eval(mod))
+        except:
+            print 'FAILURE to import ' + mod
+            output_exception()
+
+    return all_modules
+
+__all__.append('harvest_packages')
+def harvest_packages(package,ignore = None):
+    """ Retreive a list of all sub-packages that live within a package.
+
+         Only retreive packages that are immediate children of this
+         package -- do not recurse through child packages or
+         directories.  The returned list contains actual package objects, not
+         just their names.
+    """
+    join = os.path.join
+
+    d,f = os.path.split(package.__file__)
+
+    common_dir = os.path.abspath(d)
+    all_files = os.listdir(d)
+
+    all_files = remove_ignored_files(all_files,ignore,'')
+    #print 'all_files:', all_files
+    try:
+        prefix = package.__name__
+    except:
+        prefix = ''
+    all_packages = []
+    for directory in all_files:
+        path = join(common_dir,directory)
+        if os.path.isdir(path) and \
+           os.path.exists(join(path,'__init__.py')):
+            sub_package = prefix + '.' + directory
+            #print 'sub-package import ' + sub_package
+            try:
+                exec ('import ' + sub_package)
+                all_packages.append(eval(sub_package))
+            except:
+                print 'FAILURE to import ' + sub_package
+                output_exception()
+    return all_packages
+
+__all__.append('harvest_modules_and_packages')
+def harvest_modules_and_packages(package,ignore=None):
+    """ Retreive list of all packages and modules that live within a package.
+
+         See harvest_packages() and harvest_modules()
+    """
+    all = harvest_modules(package,ignore) + harvest_packages(package,ignore)
+    return all
+
+__all__.append('harvest_test_suites')
+def harvest_test_suites(package,ignore = None,level=10):
+    """
+        package -- the module to test.  This is an actual module object
+                   (not a string)
+        ignore  -- a list of module names to omit from the tests
+        level   -- a value between 1 and 10.  1 will run the minimum number
+                   of tests.  This is a fast "smoke test".  Tests that take
+                   longer to run should have higher numbers ranging up to 10.
+    """
+    suites=[]
+    test_modules = harvest_modules_and_packages(package,ignore)
+    #for i in test_modules:
+    #    print i.__name__
+    for module in test_modules:
+        if hasattr(module,'test_suite'):
+            try:
+                suite = module.test_suite(level=level)
+                if suite:
+                    suites.append(suite)
+                else:
+                    print "    !! FAILURE without error - shouldn't happen",
+                    print module.__name__
+            except:
+                print '   !! FAILURE building test for ', module.__name__
+                print '   ',
+                output_exception()
+        else:
+            try:
+                print 'No test suite found for ', module.__name__
+            except AttributeError:
+                # __version__.py getting replaced by a string throws a kink
+                # in checking for modules, so we think is a module has
+                # actually been overwritten
+                print 'No test suite found for ', str(module)
+    total_suite = unittest.TestSuite(suites)
+    return total_suite
+
+__all__.append('module_test')
+def module_test(mod_name,mod_file,level=10):
+    """*
+
+    *"""
+    #print 'testing', mod_name
+    d,f = os.path.split(mod_file)
+
+    # insert the tests directory to the python path
+    test_dir = os.path.join(d,'tests')
+    sys.path.insert(0,test_dir)
+
+    # call the "test_xxx.test()" function for the appropriate
+    # module.
+
+    # This should deal with package naming issues correctly
+    short_mod_name = string.split(mod_name,'.')[-1]
+    test_module = 'test_' + short_mod_name
+    test_string = 'import %s;reload(%s);%s.test(%d)' % \
+                  ((test_module,)*3 + (level,))
+
+    # This would be better cause it forces a reload of the orginal
+    # module.  It doesn't behave with packages however.
+    #test_string = 'reload(%s);import %s;reload(%s);%s.test(%d)' % \
+    #              ((mod_name,) + (test_module,)*3)
+    exec(test_string)
+
+    # remove test directory from python path.
+    sys.path = sys.path[1:]
+
+__all__.append('module_test_suite')
+def module_test_suite(mod_name,mod_file,level=10):
+    #try:
+        print ' creating test suite for:', mod_name
+        d,f = os.path.split(mod_file)
+
+        # insert the tests directory to the python path
+        test_dir = os.path.join(d,'tests')
+        sys.path.insert(0,test_dir)
+
+        # call the "test_xxx.test()" function for the appropriate
+        # module.
+
+        # This should deal with package naming issues correctly
+        short_mod_name = string.split(mod_name,'.')[-1]
+        test_module = 'test_' + short_mod_name
+        test_string = 'import %s;reload(%s);suite = %s.test_suite(%d)' % \
+                      ((test_module,)*3+(level,))
+        #print test_string
+        exec(test_string)
+
+        # remove test directory from python path.
+        sys.path = sys.path[1:]
+        return suite
+    #except:
+    #    print '    !! FAILURE loading test suite from', test_module, ':'
+    #    print '   ',
+    #    output_exception()
+
+
+
+__all__.append('rand')
+
+
+
+
+from scipy.base import alltrue, equal, shape, ravel, around, zeros,\
+     Float64, asarray, less_equal, array2string, less, ArrayType
+
+try:
+    from scipy.base import math
+except ImportError,msg:
+    print msg
+    import math