adding tests for training module