Added tests to setup.py