Wie gehe ich mit mehreren behauptet, in einem einzigen Python-unittest?

Stimme ich mit der vorherrschenden Meinung, dass man schreiben sollte eine Prüfmethode für jede Behauptung. Es gibt Situationen, wo Sie wollen, um zu überprüfen, mehrere Dinge in einem test-Methode. Hier ist meine Antwort, wie es zu tun:

# Works with unittest in Python 2.7
class ExpectingTestCase(unittest.TestCase):
    def run(self, result=None):
        self._result = result
        self._num_expectations = 0
        super(ExpectingTestCase, self).run(result)

    def _fail(self, failure):
        try:
            raise failure
        except failure.__class__:
            self._result.addFailure(self, sys.exc_info())

    def expect_true(self, a, msg):
        if not a:
            self._fail(self.failureException(msg))
        self._num_expectations += 1

    def expect_equal(self, a, b, msg=''):
        if a != b:
            msg = '({}) Expected {} to equal {}. '.format(self._num_expectations, a, b) + msg
            self._fail(self.failureException(msg))
        self._num_expectations += 1

Und hier sind einige Situationen, wo ich denke, es ist nützlich und nicht riskant:

1) Wenn Sie wollen test-code für verschiedene Gruppen von Daten. Hier haben wir eine add () - Funktion, und ich möchte um es zu testen mit ein paar Beispiel-Eingänge. Schreiben 3 test-Methoden für die 3 Datensätze Mittel wiederholen sich, die schlecht ist. Vor allem, wenn der Anruf war aufwändiger.:

class MyTest(ExpectingTestCase):
    def test_multiple_inputs(self):
        for a, b, expect in ([1,1,2], [0,0,0], [2,2,4]):
            self.expect_equal(expect, add(a,b), 'inputs: {} {}'.format(a,b))

2) Wenn Sie überprüfen möchten, mehrere Ausgänge einer Funktion. Ich möchte prüfen Sie jede Ausgabe, aber ich glaube nicht, dass ein Erster Fehler zu maskieren die anderen beiden.

class MyTest(ExpectingTestCase):
    def test_things_with_no_side_effects(self):
        a, b, c = myfunc()
        self.expect_equal('first value', a)
        self.expect_equal('second value', b)
        self.expect_equal('third value', c)

3) Testen mit dem schweren setup-Kosten. Tests müssen schnell laufen, oder dass die Leute aufhören, Sie zu verwenden. Einige tests erfordern, dass eine db-oder Netzwerk-Verbindung, die dauert eine Sekunde, das wäre wirklich verlangsamen Ihren test. Wenn Sie die Prüfung der db-Verbindung selbst, dann werden Sie wahrscheinlich benötigen, um die Geschwindigkeit Treffer. Aber wenn Sie etwas testen nichts zu tun haben, das wollen wir langsam das setup für eine ganze Reihe von Prüfungen durch.

InformationsquelleAutor Winston

Mit einem untertest, die Ausführung würde nicht aufhören, nach dem ersten Ausfall
https://docs.python.org/3/library/unittest.html#subtests

Hier ein Beispiel mit zwei fail behauptet:

class TestMultipleAsserts(unittest.TestCase):

    def test_multipleasserts(self):
        with self.subTest():
            self.assertEqual(1, 0)
        with self.subTest():
            self.assertEqual(2, 0)

Ausgabe:

======================================================================
FAIL: test_multipleasserts (__main__.TestMultipleAsserts) (<subtest>)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "./test.py", line 9, in test_multipleasserts
    self.assertEqual(1, 0)
AssertionError: 1 != 0

======================================================================
FAIL: test_multipleasserts (__main__.TestMultipleAsserts) (<subtest>)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "./test.py", line 11, in test_multipleasserts
    self.assertEqual(2, 0)
AssertionError: 2 != 0

----------------------------------------------------------------------
Ran 1 test in 0.000s

FAILED (failures=2)

Können Sie einfach wickeln Teilprüfungen wie folgt

class MyTestCase(unittest.TestCase):
    def expectEqual(self, first, second, msg=None):
        with self.subTest():
            self.assertEqual(first, second, msg)

class TestMA(MyTestCase):
    def test_ma(self):
        self.expectEqual(3, 0)
        self.expectEqual(4, 0)

InformationsquelleAutor Vitalii Blagodir