code: def get_test_data():
      x = np.array(
          [
              [
                  # O   B-X  I-X  B-Y  I-Y
                  [0.0, 1.0, 0.0, 0.0, 0.0],
                  [0.0, 0.0, 1.0, 0.0, 0.0],
                  [0.0, 0.0, 1.0, 0.0, 0.0],
              ],
              [
                  # O   B-X  I-X  B-Y  I-Y
                  [0.0, 1.0, 0.0, 0.0, 0.0],
                  [0.0, 1.0, 0.0, 0.0, 0.0],
                  [0.0, 1.0, 0.0, 0.0, 0.0],
              ],
          ]
      )
      y = np.array(
          [[1, 2, 2], [1, 1, 1]]
      )  # B-X  I-X  I-X  # B-X  B-X  B-X
      return x, y
  
file_path: addons/tensorflow_addons/text/tests/crf_wrapper_test.py
imports: import os
  import tempfile
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.text.crf_wrapper import CRFModelWrapper
label_desc: Use image node_id 1 for calling a global function with example usage: get_test_data() and returns: x,  y