Put tests back in setup.py