remove print sentences and replace it for assertEqual sentences