Add test to setup.py