Python Unit Testing – Using Mocks