Add test files to setup.py