Adding the basic tests in the tests.py file.