Create tests.py

unit tests