Using a custom error class

[germa89]

This is useful when used together with pytest-rerunfailures, and it can help to also with try context managers.