I know not every test uses this, but is there a reason otherwise to not just put this in `__init__.py`?