为什么cProfile模块不能与unittest一起使用?

时间:2012-07-25 08:06:14

标签: python macos unit-testing cprofile

我会使用cProfile模块来分析我的单元测试。但是当我跑步时

python -mcProfile mytest.py

我在'0.000s'中得到了'Ran 0测试'。这是mytest.py

的源代码
import unittest

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

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

我也测试了其他更复杂的单元测试。如果我用cProfile运行它,总是得到'Ran 0 tests'。请帮忙。

更新:我的操作系统是内置python 2.7的MacOS 10.7。相同的代码在ubuntu上正常工作。

3 个答案:

答案 0 :(得分:12)

您必须在测试的构造函数中初始化cProfiler,并在析构函数中使用配置文件数据 - 我这样使用它:

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

如果测试等待输入,则需要在该调用之前调用self.pr.disable(),然后重新启用它。

答案 1 :(得分:2)

我不确定为什么,但是显式创建和运行测试套件似乎可行。我添加了time.sleep(2),以显示统计信息中的确定性内容。

import time
import unittest

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

if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(TestBasic)
    unittest.TextTestRunner(verbosity=2).run(suite)

以bash运行,仅保留前10行,我们可以看到{time.sleep}是运行时间最长的呼叫:

~ $ python -m cProfile -s tottime so.py | head -n10
testFoo (__main__.TestBasic) ... ok

----------------------------------------------------------------------
Ran 1 test in 2.003s

OK
         1084 function calls (1072 primitive calls) in 2.015 seconds

   Ordered by: internal time

   ncalls  tottime  percall  cumtime  percall filename:lineno(function)
        1    2.003    2.003    2.003    2.003 {time.sleep}
        1    0.002    0.002    0.003    0.003 case.py:1(<module>)
        1    0.002    0.002    0.003    0.003 collections.py:11(<module>)
        1    0.001    0.001    2.015    2.015 so.py:1(<module>)
        1    0.001    0.001    0.010    0.010 __init__.py:45(<module>)

答案 2 :(得分:2)

指定模块明确为我解决了这个问题。也就是说,替换...

    unittest.main()

...(试图自动发现应该运行哪些测试)

    unittest.main(module='mytest')  # if in 'mytest.py', or as appropriate for your filename/module

...允许正确配置为python -m cProfile mytest.py