Mock A Function In Python