blob: c0845b610c6123b2a4b3c961cca188d982f1b68d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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
|
