Update trainpredicttest.py