Pytest Mock A Function