From 28577aae928847e64a0274b5922e26e1f15d9916 Mon Sep 17 00:00:00 2001 From: Jelmer Vernooij Date: Fri, 8 Jan 2010 02:09:20 +0100 Subject: Import testtools as well, required for subunit. --- lib/subunit/python/testtools/utils.py | 39 +++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 lib/subunit/python/testtools/utils.py (limited to 'lib/subunit/python/testtools/utils.py') 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 -- cgit