Update setup.py

Making `python setup.py test` work by adding test_suite.