Comment by taco_emoji

You could just use nested classes for that, with the added benefit that the test framework will reflect that organization in its UX.