Как я могу извлечь список TestCases из TestSuite?

Я использую Python unittest с таким простым кодом, как этот:

suite = unittest.TestSuite()
suite.addTest(unittest.defaultTestLoader.loadTestsFromTestCase(module1))
suite.addTest(unittest.defaultTestLoader.loadTestsFromTestCase(module2))

Однако я хочу сделать некоторые пользовательские вещи для каждого теста после того, как они были собраны набором. Я думал, что могу сделать что-то вроде этого, чтобы перебрать тестовые примеры в наборе:

print suite.countTestCases()
for test in suite:             # Also tried with suite.__iter__()
    # Do something with test
    print test.__class__

Однако для того количества тестовых примеров, которое я загружаю, он всегда печатает только

3
<class 'unittest.suite.TestSuite'>

Есть ли способ получить все объекты класса TestCase из пакета? Есть ли другой способ загрузки тестовых примеров, чтобы облегчить это?

6
задан elynnaie 25 January 2012 в 14:38
поделиться