additional tests and 1st step to refactor epoche trainer