Change a Function's Output Based on Input Parameter Value in a Python Unit Test - Thomas Stringer