Python Peeps: For Unit Testing when do you prefer to use Unittest, Doctest, Pytest ?

Nemesis Contreras — Fri, 20 Apr 2018 02:46:53 +0000

Examples of Unittest , Doctest and Pytest are listed bellow-respectively- if you aren't familiar with them.

# Testing Framework:
# Unittest supports test automation by writing classes.

import unittest


def fib(n):
    """Return the n-th number in the fibonacci series."""
    if n <= 0:
        return 0
    elif n == 1:
        return 1
    return fib(n - 1) + fib(n - 2)


class TestFib(unittest.TestCase):

    def test_fib_series(self):
        self.assertEqual(fib(0), 0)
        self.assertEqual(fib(1), 1)
        self.assertEqual(fib(2), 1)
        self.assertEqual(fib(3), 2)
        self.assertEqual(fib(4), 3)
        self.assertEqual(fib(5), 5)


if __name__ == "__main__":
    # normally this is unittest.main()...this is for jupyter
    unittest.main(argv=['first-arg-is-ignored'], exit=False, verbosity=2)

# Testing Framework:
# Doctest allows you to write tests in the docstrings

def fib(n):
    """Return the n-th number in the fibonacci series.

    >>> fib(0)
    0
    >>> fib(1)
    1
    >>> fib(29)
    514229
    """
    if n <= 0:
        return 0
    elif n == 1:
        return 1
    return fib(n - 1) + fib(n - 2)


if __name__ == "__main__":
    import doctest

    # doctest.testmod()
    doctest.testmod(verbose=True)

#Testing framework pytest
# Test discovery: for doctests, unittests and pytests.

import pytest


%%run_pytest[clean] -vvv

def fib(n):
    """Return the n-th number in the fibonacci series."""
    if n <= 0:
        return 0
    elif n == 1:
        return 1
    return fib(n-1) + fib(n-2)

def test_fib_series():
    assert fib(0) == 0
    assert fib(1) == 1
    assert fib(2) == 1
    assert fib(3) == 2
    assert fib(4) == 3
    assert fib(5) == 5

DEV Community: Nemesis Contreras

Python Peeps: For Unit Testing when do you prefer to use Unittest, Doctest, Pytest ?