code: def test_get_leaf_sample_counts(xgb_tree):
      leaf_ids, leaf_samples = xgb_tree.get_leaf_sample_counts()
      assert np.array_equal(
          leaf_ids, np.array([3, 5, 6, 2])
      ), "Leaf ids should be [3, 5, 6, 2]"
      assert np.array_equal(
          leaf_samples, np.array([7, 9, 3, 1])
      ), "Leaf samples should be [7, 9, 3, 1]"
  
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 8 for calling a global function with example usage: test_get_leaf_sample_counts(xgb_tree) without return types