code: def test_session_data_for_ids(model_data: RasaModelData):
      filtered_data = model_data._data_for_ids(
          model_data.data, np.array([0, 1])
      )
  
      for values in filtered_data.values():
          for data in values.values():
              for v in data:
                  assert np.array(v).shape[0] == 2
  
      key = model_data.keys()[0]
      sub_key = model_data.keys(key)[0]
  
      assert np.all(
          np.array(filtered_data[key][sub_key][0][0])
          == np.array(model_data.get(key, sub_key)[0][0])
      )
      assert np.all(
          np.array(filtered_data[key][sub_key][0][1])
          == np.array(model_data.get(key, sub_key)[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 6 for calling a global function with example usage: test_session_data_for_ids(model_data) without return types