python - ¿Por qué el módulo cProfile no funciona con unittest?

macos unit-testing

Question

Yo usaría el módulo cProfile para perfilar mis pruebas unitarias. Pero cuando corro

python -mcProfile mytest.py

Tengo ''Ran 0 pruebas en 0.000s''. Aquí está el código fuente de mytest.py

import unittest

class TestBasic(unittest.TestCase):
    def testFoo(self):
        assert True == True

if __name__ == ''__main__'':
    unittest.main()

También he probado con otras pruebas de unidad más complejas. Siempre tengo ''Ran 0 tests'' si lo ejecuto con cProfile. Por favor ayuda.

ACTUALIZACIÓN: Mi sistema operativo es MacOS 10.7 con python incorporado 2.7. El mismo código funciona correctamente en ubuntu.

Answer 1

Tienes que inicializar el cProfiler en el Constructor de la prueba, y usar los datos del perfil en el destructor. Lo uso así:

from pstats import Stats
import unittest

class TestSplayTree(unittest.TestCase):
    """a simple test"""

def setUp(self):
    """init each test"""
    self.testtree = SplayTree (1000000)
    self.pr = cProfile.Profile()
    self.pr.enable()
    print "/n<<<---"

def tearDown(self):
    """finish any test"""
    p = Stats (self.pr)
    p.strip_dirs()
    p.sort_stats (''cumtime'')
    p.print_stats ()
    print "/n--->>>"
def xtest1 (self):
    pass

En caso de que una prueba espere la entrada, uno debe llamar a self.pr.disable() antes de esa llamada y volver a habilitarla después.