You should be able to just patch the `TestCase` class with this to save a few lines without much overhead.