Python Unittest for only one function

You have to write a python unit test for only one function. if your code looks good then we ll continue to build the rest of the function.