code: def test_get_node_nsamples_by_class(tf_shadow_clf):
      assert tf_shadow_clf.get_node_nsamples_by_class(0) == (549, 342)
      assert tf_shadow_clf.get_node_nsamples_by_class(1) == (468, 109)
      assert tf_shadow_clf.get_node_nsamples_by_class(4) == (81, 233)
  
imports: import pytest
  import numpy
  import tensorflow_decision_forests
  from dtreeviz.models.tensorflow_decision_tree import ShadowTensorflowTree
label_desc: Use image node_id 14 for calling a global function with example usage: test_get_node_nsamples_by_class(tf_shadow_clf) without return types