diff options
Diffstat (limited to 'lib/subunit/python/testtools/utils.py')
-rw-r--r-- | lib/subunit/python/testtools/utils.py | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/lib/subunit/python/testtools/utils.py b/lib/subunit/python/testtools/utils.py new file mode 100644 index 0000000000..325572297b --- /dev/null +++ b/lib/subunit/python/testtools/utils.py @@ -0,0 +1,39 @@ +# Copyright (c) 2008 Jonathan M. Lange. See LICENSE for details. + +"""Utilities for dealing with stuff in unittest.""" + + +import sys + +__metaclass__ = type +__all__ = [ + 'iterate_tests', + ] + + +if sys.version_info > (3, 0): + def _u(s): + """Replacement for u'some string' in Python 3.""" + return s + def _b(s): + """A byte literal.""" + return s.encode("latin-1") + advance_iterator = next +else: + def _u(s): + return unicode(s, "latin-1") + def _b(s): + return s + advance_iterator = lambda it: it.next() + + +def iterate_tests(test_suite_or_case): + """Iterate through all of the test cases in `test_suite_or_case`.""" + try: + suite = iter(test_suite_or_case) + except TypeError: + yield test_suite_or_case + else: + for test in suite: + for subtest in iterate_tests(test): + yield subtest |