Title: Simple Unit Test
Slug: simple_unit_test
Summary: A simple unit test in Python. Date: 2016-01-23 12:00
Category: Python
Tags: Testing
Authors: Chris Albon

Interesting in learning more? Here are some good books on unit testing in Python: Python Testing: Beginner's Guide and Python Testing Cookbook.

Preliminaries



In [1]:

    
import unittest
import sys

Create Function To Be Tested



In [2]:

    
def multiply(x, y):
    return x * y

Create Test

Note: It is standard practice to name a unit test test_ + <function being tested>. This naming standard allows for automated test using some libraries.



In [3]:

    
# Create a test case
class TestMultiply(unittest.TestCase):
    # Create the unit test
    def test_multiply_two_integers_together(self):
        # Test if 4 equals the output of multiply(2,2)
        self.assertEqual(4, multiply(2,2))

Run Test



In [4]:

    
# Run the unit test (and don't shut down the Jupyter Notebook)
unittest.main(argv=['ignored', '-v'], exit=False)









    



test_multiply_two_integers_together (__main__.TestMultiply) ... ok

----------------------------------------------------------------------
Ran 1 test in 0.001s

OK






    Out[4]:





<unittest.main.TestProgram at 0x104919128>