added unit tests to test.py