yap23cp (aka. Yest Another Python 2.3 compatiblity problem

1 files changed, 5 insertions, 1 deletions

diff --git a/testlib.py b/testlib.py
index a835889..e3b3b8a 100644
--- a/testlib.py
+++ b/testlib.py
@@ -457,7 +457,11 @@ class NonStrictTestLoader(unittest.TestLoader):
         except AttributeError:
             print "No such suite", suitename
             return []
-        return suite
+        assert hasattr(suite, '_tests'), \
+               "%s.%s is not a valid TestSuite" % (module.__name__, suitename)
+        # python2.3 does not implement __iter__ on suites, we need to return
+        # _tests explicitly
+        return suite._tests
     
         
     def loadTestsFromName(self, name, module=None):