adding more tests for training module
combining tests for training module