From: Brett Cannon <bcannon@gmail.com>
Date: Sun, 1 Feb 2009 03:51:54 +0000 (+0000)
Subject: Do not execute the .pyc/.pyo files as well as the .py files.
X-Git-Tag: v3.1a1~327
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=ae9ad186d058c5700d0692e2f3b026e95639f5cf;p=thirdparty%2FPython%2Fcpython.git

Do not execute the .pyc/.pyo files as well as the .py files.
---

diff --git a/Lib/importlib/test/__init__.py b/Lib/importlib/test/__init__.py
index f9dd57e7ec27..bda33e69fea2 100644
--- a/Lib/importlib/test/__init__.py
+++ b/Lib/importlib/test/__init__.py
@@ -9,7 +9,8 @@ def test_suite(package=__package__, directory=os.path.dirname(__file__)):
         if name.startswith('.'):
             continue
         path = os.path.join(directory, name)
-        if os.path.isfile(path) and name.startswith('test_'):
+        if (os.path.isfile(path) and name.startswith('test_') and
+                name.endswith('.py')):
             submodule_name = os.path.splitext(name)[0]
             module_name = "{0}.{1}".format(package, submodule_name)
             __import__(module_name, level=0)
@@ -20,6 +21,8 @@ def test_suite(package=__package__, directory=os.path.dirname(__file__)):
             __import__(package_name, level=0)
             package_tests = getattr(sys.modules[package_name], 'test_suite')()
             suite.addTest(package_tests)
+        else:
+            continue
     return suite