26

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

python -mcProfile mytest.py

我得到了“在 0.000 秒内运行 0 次测试”。这是 mytest.py 的源代码

import unittest

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

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

我也用其他更复杂的单元测试进行了测试。如果我使用 cProfile 运行它，总是得到“运行 0 次测试”。请帮忙。

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

4

3 回答 3

17

您必须在测试的构造函数中初始化 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()之前调用，然后重新启用它。

如果您更喜欢pytest方法名称有点不同：

import cProfile
import time

class TestProfile:
    """ test to illustrate cProfile usage """

    def setup_class(self):
        self.pr = cProfile.Profile()
        self.pr.enable()

    def teardown_class(self):
        self.pr.disable()
        self.pr.print_stats(sort="tottime")

    def test_sleep(self):
        time.sleep(2)
于 2013-11-27T19:29:17.423 回答
6

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

    unittest.main()

...（它试图自动发现它应该运行哪些测试）...

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

...允许正确分析为python -m cProfile mytest.py.

于 2020-05-15T21:30:49.167 回答
4

我不确定为什么，但明确地创建和运行测试套件似乎有效。我添加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>)
于 2019-03-15T04:15:04.537 回答