Started testing

Used mock from PyPI.