unittestで強制終了( sys.exit() )をどのように表記すればよいのか分かりません。

Question

簡略して書きますと、整数の和を得る、というtashizan関数についてunittestを用いて書いています。

def tashizan(a,b):
    try:
        return int(a)+int(b)
    except ValueError:
        print("a,bは整数ではなく文字列の可能性があります")
        sys.exit(1)

関数を見る通り、引数にINTEGERではなく文字列が渡されたとき、スクリプトを止めたいためexcept ValueErrorで標準出力した後にsys.exit()でスクリプトを止めます。これをunittestで書いた場合以下のように書きました。関数の書き方及びunittestの書き方にアドバイスが有れば指摘してほしいです。現状ではFAILED (errors=1)になってしまいます。異常な引数を受け取ったらスクリプトを止める、かつ、unittestでFAILEDが表示されないことを目指しています。

import unittest , tashizan

class TestTashizan(unittest.TestCase):

    def test_tashizan_1(self):
        expected = 8
        actual = tashizan(2, 6)
        self.assertEqual(expected, actual)

    def test_tashizan_2(self):
        expected = 8
        actual = tashizan(6, "2")
        self.assertEqual(expected, actual)


    def test_tashizan_3(self):
        actual = tashizan("a", "6")
        self.assertRaises(ValueError)


if __name__ == "__main__":
    unittest.main(exit=False)

結果は以下のようになります。

..a,bは整数ではなく文字列の可能性があります
E
======================================================================
ERROR: test_tashizan_3 (__main__.TestTashizan)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "C:\Users\USER\Desktop\function.py", line 7, in tashizan
    return int(a) + int(b)
ValueError: invalid literal for int() with base 10: 'a'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "unit_test.py", line 21, in test_tashizan_3
    actual = tashizan("a", "6")
  File "C:\Users\USER\Desktop\function.py", line 10, in tashizan
    sys.exit(1)
SystemExit: 1

----------------------------------------------------------------------
Ran 3 tests in 0.001s

FAILED (errors=1)

なお関数のsys.exit()をコメントアウトするとFAILEDは表示されずうまくいくのですが、本来の目的であるスクリプトを止めるのが実現できなくなってしまいます。

Answer 1

sys.exitは例外SystemExitを投げるので、それを監視してやれば良いです。
またassertRaisesの使い方も間違ってます。

手元の環境ではこれでテストが通りました。

def test_tashizan_3(self):
    with self.assertRaises(SystemExit):
        tashizan("a", "6")

あるいは

def test_tashizan_3(self):
    self.assertRaises(SystemExit, lambda: tashizan("a", "6"))

参考： unittest --- ユニットテストフレームワーク — Python 3.7.4 ドキュメント

Comments

[chapin]

ありがとうございます！今日一日中調べてもうまく行かなかったので大変助かりました。
本当にありがとうございます！