I added mock to the tests and mock is a standard module in Python 3 so I too think mock is the way to go.