Update setup.py
Add test dataset