Refactor tests to use BallotGenerator.