Add tests to setup.py