Added tests.py
Added tests.py