Add the test to ensure "INFO: Done training" is in output.