web-dev-qa-db-ja.com

Python単体テスト：テストが失敗したときにデバッガを自動的に実行する

単体テストが失敗した時点でデバッガーを自動的に起動する方法はありますか？

現在、私は手動でpdb.set_trace（）を使用していますが、毎回追加して最後に取り出す必要があるため、これは非常に面倒です。

例えば：

import unittest

class tests(unittest.TestCase):

    def setUp(self):
        pass

    def test_trigger_pdb(self):
        #this is the way I do it now
        try:
            assert 1==0
        except AssertionError:
            import pdb
            pdb.set_trace()

    def test_no_trigger(self):
        #this is the way I would like to do it:
        a=1
        b=2
        assert a==b
        #magically, pdb would start here
        #so that I could inspect the values of a and b

if __name__=='__main__':
    #In the documentation the unittest.TestCase has a debug() method
    #but I don't understand how to use it
    #A=tests()
    #A.debug(A)

    unittest.main()
pythonunit-testingpdbpython-unittest
42
tjb
import unittest
import sys
import pdb
import functools
import traceback
def debug_on(*exceptions):
    if not exceptions:
        exceptions = (AssertionError, )
    def decorator(f):
        @functools.wraps(f)
        def wrapper(*args, **kwargs):
            try:
                return f(*args, **kwargs)
            except exceptions:
                info = sys.exc_info()
                traceback.print_exception(*info) 
                pdb.post_mortem(info[2])
        return wrapper
    return decorator

class tests(unittest.TestCase):
    @debug_on()
    def test_trigger_pdb(self):
        assert 1 == 0

Set_traceの代わりに例外でpost_mortemを呼び出すようにコードを修正しました。

24
Rosh Oxymoron

あなたが探しているのは 鼻 だと思います。 nittest のテストランナーのように機能します。

次のコマンドを使用して、エラー時にデバッガーにドロップできます。

nosetests --pdb
36
cmcginty

単純なオプションは、結果を収集せずにテストを実行し、最初の例外でスタックをクラッシュさせることです（任意の事後分析処理の場合）。

_try: unittest.findTestCases(__main__).debug()
except:
    pdb.post_mortem(sys.exc_info()[2])
_

別のオプション：デバッグテストランナーで_unittest.TextTestResult_のaddErrorとaddFailureをオーバーライドして、即座にpost_mortemデバッグする（tearDown()の前に）-またはエラーとトレースバックを収集して処理する高度な方法で。

（追加のフレームワークやテストメソッド用の追加のデコレータは必要ありません）

基本的な例：

_import unittest, pdb

class TC(unittest.TestCase):
    def testZeroDiv(self):
        1 / 0

def debugTestRunner(post_mortem=None):
    """unittest runner doing post mortem debugging on failing tests"""
    if post_mortem is None:
        post_mortem = pdb.post_mortem
    class DebugTestResult(unittest.TextTestResult):
        def addError(self, test, err):
            # called before tearDown()
            traceback.print_exception(*err)
            post_mortem(err[2])
            super(DebugTestResult, self).addError(test, err)
        def addFailure(self, test, err):
            traceback.print_exception(*err)
            post_mortem(err[2])
            super(DebugTestResult, self).addFailure(test, err)
    return unittest.TextTestRunner(resultclass=DebugTestResult)

if __name__ == '__main__':
    ##unittest.main()
    unittest.main(testRunner=debugTestRunner())
    ##unittest.main(testRunner=debugTestRunner(pywin.debugger.post_mortem))
    ##unittest.findTestCases(__main__).debug()
_
4
kxr

サードパーティのテストフレームワークの拡張には、一般的に機能（noseおよびnose2は他の回答ですでに言及されています）。さらにいくつか：

pytest はそれをサポートしています。

pytest --pdb

または、absltestモジュールの代わりに absl-py のunittestを使用する場合：

name_of_test.py --pdb_post_mortem
1
gps

@ cmcginty's answer を後継者に適用するには nose 2 （ 鼻で推奨 Debianベースのシステムでapt-get install nose2）を介して使用できます デバッガにドロップ を呼び出して失敗やエラーが発生します

nose2

テストディレクトリ。

そのためには、ホームディレクトリに適切な.unittest.cfg、またはプロジェクトディレクトリにunittest.cfgが必要です。行を含める必要があります

[debugger]
always-on = True
errors-only = False
0
serv-inc

これが組み込みの追加モジュールなしのソリューションです。

import unittest
import sys
import pdb

####################################
def ppdb(e=None):
    """conditional debugging
       use with:  `if ppdb(): pdb.set_trace()` 
    """
    return ppdb.enabled

ppdb.enabled = False
###################################


class SomeTest(unittest.TestCase):

    def test_success(self):
        try:
            pass
        except Exception, e:
            if ppdb(): pdb.set_trace()
            raise

    def test_fail(self):
        try:
            res = 1/0
            #note:  a `nosetests --pdb` run will stop after any exception
            #even one without try/except and ppdb() does not not modify that.
        except Exception, e:
            if ppdb(): pdb.set_trace()
            raise


if __name__ == '__main__':
    #conditional debugging, but not in nosetests
    if "--pdb" in sys.argv:
        print "pdb requested"
        ppdb.enabled = not sys.argv[0].endswith("nosetests")
        sys.argv.remove("--pdb")

    unittest.main()

python myunittest.py --pdbで呼び出すと停止します。そうでなければそれはしません。

0
JL Peyret