Mocking python function based on input arguments Mocking python function based on input arguments python python