Prepare setup.py to run tests