code: def test_get_node_nsamples_by_class(spark_dtree):
      assert np.array_equal(
          spark_dtree.get_node_nsamples_by_class(0),
          np.array([549.0, 342.0]),
      )
      assert np.array_equal(
          spark_dtree.get_node_nsamples_by_class(1),
          np.array([468.0, 109.0]),
      )
      assert np.array_equal(
          spark_dtree.get_node_nsamples_by_class(5),
          np.array([463.0, 96.0]),
      )
      assert np.array_equal(
          spark_dtree.get_node_nsamples_by_class(8),
          np.array([1.0, 2.0]),
      )
      assert np.array_equal(
          spark_dtree.get_node_nsamples_by_class(11),
          np.array([8.0, 159.0]),
      )
      assert np.array_equal(
          spark_dtree.get_node_nsamples_by_class(13),
          np.array([51.0, 70.0]),
      )
      assert np.array_equal(
          spark_dtree.get_node_nsamples_by_class(16),
          np.array([21.0, 2.0]),
      )
  
imports: import numpy
  import pyspark
  import pytest
  from pyspark.ml.classification import DecisionTreeClassificationModel
  from pyspark.sql import SparkSession
  from dtreeviz.models.spark_decision_tree import ShadowSparkTree
label_desc: Use image node_id 11 for calling a global function with example usage: test_get_node_nsamples_by_class(spark_dtree) without return types