Mocking A Function In Pytest Example