Add tests for trainings