Search⌘ K
Home
Courses
Python 201 - Interactively Learn Advanced Concepts in Python 3

A Simple Example of unittest Module

Explore how to use Python's unittest module to write and run simple test cases that validate functions. Understand how to organize tests, use assertions, and interpret test results to ensure your code behaves as expected.

We'll cover the following...

We always find a code example or two to be the quickest way to learn how something new works. So let’s create a little module that we will call mymath.py. Then put the following code into it:

Python 3.5
def add(a, b):
    return a + b
def subtract(a, b):
    return a - b
def multiply(a, b):
    return a * b
def divide(numerator, denominator):
    return float(numerator) / denominator
print(divide(34,5))
print(add(4,5))
print(subtract(3,4))
print(multiply(3,4))

This module defines four mathematical functions: add, subtract, multiply and divide. They do not do any error checking and they actually don’t do exactly what we might expect. For example, if we were to call the add function with two strings, it would happily concatenate ...