adding unit test file
corrected tests.py