Python Unittest Tutorial | Unit Testing in Python using unittest Framework

Unit testing checks small pieces of code (like functions or classes) to confirm they work correctly. In Python, this is done with the unittest framework, which is built into the standard library and follows the xUnit style. The unittest module is widely used because:

• No setup needed: comes pre-installed with Python.
• Structured: supports test cases, fixtures, assertions, and test suites.
• Automated: can discover and run tests across the project.
• Stable and versatile: reliable for projects of any size, with easy integration into other tools.

This makes unittest a dependable choice for creating systematic and automated tests in Python.

Assert Methods in Unittest Framework

unittest has many methods to assert on the values, types and existence of variables. Below are some of the methods that are commonly used to write assertions:

OOP Concepts Supported by Unittest Framework

The White Box Testing method is used for Unit tests. Below are some of supported oops concept by Unitttest framework:

• test fixture: A test fixture is used as a baseline for running tests to ensure that there is a fixed environment in which tests are run so that results are repeatable. Examples: creating temporary databases or starting a server process.
• test case: A test case is a set of conditions which is used to determine whether a system under test works correctly.
• test suite: Test suite is a collection of testcases that are used to test a software program to show that it has some specified set of behaviours by executing the aggregated tests together.
• test runner: A test runner is a component which set up the execution of tests and provides the outcome to the user.

Python Unittest Example

Example 1: We will write a simple function add() that returns the sum of two numbers. Using unittest framework, we create a test case class TestAddFunction to verify different scenarios of this function.

#test.py 
import unittest

def add(a, b):
    return a + b

class TestAddFunction(unittest.TestCase):
    def test_add_positive_numbers(self):
        self.assertEqual(add(1, 2), 3)

    def test_add_negative_numbers(self):
        self.assertEqual(add(-1, -2), -3)

    def test_add_mixed_numbers(self):
        self.assertEqual(add(1, -2), -1)
        self.assertEqual(add(-1, 2), 1)

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

To run the unit tests, run the following command in your terminal:

python test.py

Output

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

OK

If a test fails (e.g., expected value is changed), output shows the failed test:

..F
======================================================================
FAIL: test_add_positive_numbers (__main__.TestAddFunction)
----------------------------------------------------------------------
Traceback (most recent call last):
File "f:\GeeksforGeeks\example_package\main.py", line 8, in test_add_positive_numbers
self.assertEqual(add(1, 2), 2)
AssertionError: 3 != 2

Ran 3 tests in 0.001s
FAILED (failures=1)

Use -v option for detailed results:

python test.py -v

Output

test_add_mixed_numbers (__main__.TestAddFunction) ... ok
test_add_negative_numbers (__main__.TestAddFunction) ... ok
test_add_positive_numbers (__main__.TestAddFunction) ... ok

Ran 3 tests in 0.002s
OK

Outcomes Possible in Unit Testing

• OK -> All tests passed.
• FAIL -> Test assertion failed (AssertionError).
• ERROR -> Test raised an unexpected exception.

Example 2: This example demonstrates testing different string operations using unittest framework. The test case class TestStringMethods includes multiple tests to verify string properties and behavior.

import unittest

class TestStringMethods(unittest.TestCase):

    def setUp(self):
        pass

    # Returns True if the string contains 4 a.
    def test_strings_a(self):
        self.assertEqual('a'*4, 'aaaa')

    # Returns True if the string is in upper case.
    def test_upper(self):
        self.assertEqual('foo'.upper(), 'FOO')

    def test_isupper(self):
        self.assertTrue('FOO'.isupper())
        self.assertFalse('Foo'.isupper())

    def test_strip(self):
        s = 'geeksforgeeks'
        self.assertEqual(s.strip('geek'), 'sforgeeks')

    # Returns true if the string splits and matches
    # the given output.
    def test_split(self):
        s = 'hello world'
        self.assertEqual(s.split(), ['hello', 'world'])
        with self.assertRaises(TypeError):
            s.split(2)

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

Output

.....
----------------------------------------------------------------------
Ran 5 tests in 0.000s

OK

Explanation:

• assertEqual(): Checks if the result equals the expected value.
• assertTrue() / assertFalse(): Verifies if a condition is True or False.
• assertRaises(): Confirms a specific exception is raised.
• test_strings_a(): Verifies character multiplication.
• test_upper(): Checks string conversion to uppercase.
• test_isupper(): Checks uppercase property of string.
• test_strip(): Ensures specified characters are removed from string.
• test_split(): Checks splitting of string and validates TypeError for invalid input.
• unittest.main(): Runs all tests and provides a command-line interface.