Make testable using python setup.py test