How To Mock A Function In Python