Python Mock Function To Return Value