code: def test_balance_model_data(model_data: RasaModelData):
      data = model_data.balanced_data(model_data.data, 2, False)
  
      assert np.all(
          np.array(data["label"]["ids"][0]) == np.array([0, 1, 1, 0, 1])
      )
  
file_path: rasa/tests/utils/tensorflow/test_model_data.py
imports: import copy
  import pytest
  import numpy
  from rasa.utils.tensorflow.model_data import RasaModelData
label_desc: Use image node_id 10 for calling a global function with example usage: test_balance_model_data(model_data) without return types