Finished added tests to test.py.