CLN: Remove classes from tests.apply
- [x] Ensure all linting tests pass, see [here](https://pandas.pydata.org/pandas-docs/dev/development/contributing.html#code-standards) for how to run them

No changes to the tests, other than to remove the unused class structure.