code: def test_get_prediction(xgb_tree):
      assert xgb_tree.get_prediction(0) == 28.254867647058823
      assert xgb_tree.get_prediction(1) == 29.531439628482975
      assert xgb_tree.get_prediction(2) == 4.0
      assert xgb_tree.get_prediction(3) == 40.52844537815126
      assert xgb_tree.get_prediction(4) == 23.116519607843134
      assert xgb_tree.get_prediction(5) == 26.04424836601307
      assert xgb_tree.get_prediction(6) == 14.333333333333334
  
file_path: dtreeviz/testing/testlib/models/test_decision_tree_xgb_regressor.py
imports: import joblib
  import numpy
  import pytest
  import xgboost
  from dtreeviz.models.xgb_decision_tree import ShadowXGBDTree
label_desc: Use image node_id 6 for calling a global function with example usage: test_get_prediction(xgb_tree) without return types