add test on trained model