code: def make_imag_tests(options):
      """Make a set of tests to do imag op."""
  
      # Chose a set of parameters
      test_parameters = [
          {
              "dtype": [tf.complex64, tf.complex128],
              "input_shape": [
                  [],
                  [1],
                  [2, 3],
                  [1, 3, 4, 3],
                  [2, 2, 3, 4, 5, 6],
              ],
          }
      ]
  
      def build_graph(parameters):
          input_tensor = tf.compat.v1.placeholder(
              dtype=parameters["dtype"],
              name="input",
              shape=parameters["input_shape"],
          )
          out = tf.math.imag(input_tensor)
          return [input_tensor], [out]
  
      def build_inputs(parameters, sess, inputs, outputs):
          input_values = create_tensor_data(
              parameters["dtype"].as_numpy_dtype,
              parameters["input_shape"],
              min_value=-10,
              max_value=10,
          )
          return [input_values], sess.run(
              outputs, feed_dict=dict(zip(inputs, [input_values]))
          )
  
      make_zip_of_tests(
          options, test_parameters, build_graph, build_inputs
      )
  
imports: import tensorflow
  from tensorflow.lite.testing.zip_test_utils import create_tensor_data
  from tensorflow.lite.testing.zip_test_utils import make_zip_of_tests
  from tensorflow.lite.testing.zip_test_utils import register_make_test_function
label_desc: Use image node_id 1 for calling a global function with example usage: make_imag_tests(options) and returns: list,  list, list,  sess