Add Mock to setup.py