Create test_xgboost.py