test_models function

Test machine learning models on test data