One way to mock a function is to use the create_autospec function, which will mock out an object according to its specs. With functions, we can use this to ensure that they are called appropriately.
With a function multiply in custom_math.py:
def multiply(a, b):
return a * b
And a function...