If you fail your first attempt at the train driver assessment, you're permitted a second attempt provided it is more than six months after the first. Regardless of which particular sections you failed on initially, you will be required to sit the entire test suite again.