add testcase for Train and Classify