Using a python context manager to time your code is a very nice way to do this.

honestly, so, you do not go to rewrite the functions or pieces of code.

good example also in go.