Adds test support in setup.py