Python Mock Return Value Based On Input