code: def test_get_prediction(xgb_tree):
      assert xgb_tree.get_prediction(3) == 1
      assert xgb_tree.get_prediction(4) == 1
      assert xgb_tree.get_prediction(5) == 0
      assert xgb_tree.get_prediction(6) == 0
  
imports: import joblib
  import numpy
  import pytest
  import xgboost
  from dtreeviz.models.xgb_decision_tree import ShadowXGBDTree
label_desc: Use image node_id 12 for calling a global function with example usage: test_get_prediction(xgb_tree) without return types