Pytest Mock Example