code: def test_local_predict_dataframe(setup):
      dtrain = MarsDMatrix(X_df, y_series)
      booster = train({}, dtrain, num_boost_round=2)
      assert isinstance(booster, Booster)
  
      prediction = predict(booster, X_df)
      assert isinstance(prediction.to_pandas(), pd.Series)
  
file_path: mars/mars/learn/contrib/xgboost/tests/test_predict.py
imports: import numpy
  import pandas
  import pytest
  from .....None import tensor
  from .....None import dataframe
  from ..None import MarsDMatrix, train, predict
label_desc: Use image node_id 2 for calling a global function with example usage: test_local_predict_dataframe(setup) without return types