Created test.py file and started unit testing.