Following the pattern that we try to apply in other test files, there should be a class per function. In this case 3 new classes for these 4 new tests. I know that this file doesn't follow this pattern though, so feel free to leave as is.