Create tests.py

This file contain basic test method for the ease of future development.