Create unit-test.py

This code is to cover basics of unittest.