Assign generator unit tests to base classes