pytestconf.py
changeset 4804 daa71eaf11e8
parent 4796 a20edc0f8b30
child 5158 5e9055b8c10a
child 5421 8167de96c523
--- a/pytestconf.py	Thu Mar 04 12:08:23 2010 +0100
+++ b/pytestconf.py	Thu Mar 04 17:51:19 2010 +0100
@@ -13,7 +13,12 @@
             return super(CustomPyTester, self).testfile(filename, batchmode)
         finally:
             modname = splitext(split(filename)[1])[0]
-            for cls in vars(sys.modules[modname]).values():
+            try:
+                module = sys.modules[modname]
+            except KeyError:
+                # error during test module import
+                return
+            for cls in vars(module).values():
                 if getattr(cls, '__module__', None) != modname:
                     continue
                 clean_repo_test_cls(cls)