Create unitest.py

Unitary tests