Enable running tests through setup.py
Also install python mock library for tests