I think we should keep a copy of this file as example file, and it can be the basis for a notebook for the documentation.

The test files needs to have everything inside a test function or test class, without savefig