code: def test_if_unary_double_if_positive_else_square(
      self, compute_unit, backend
  ):
      @make_tf_graph([(1,)])
      def build_model(x):
          if x >= 0:
              out = x + x
          else:
              out = x * x
          return out
  
      model, inputs, outputs = build_model
      input_values = [np.array([2], dtype=np.float32)]
      input_dict = dict(zip(inputs, input_values))
      TensorFlowBaseTest.run_compare_tf(
          model,
          input_dict,
          outputs,
          compute_unit=compute_unit,
          backend=backend,
      )
  
imports: import itertools
  import numpy
  import pytest
  import coremltools
  from coremltools.converters.mil import testing_reqs
  from coremltools.converters.mil.frontend.tensorflow.test.testing_utils import TensorFlowBaseTest
  from coremltools.converters.mil.frontend.tensorflow2.test.testing_utils import TensorFlow2BaseTest
  from coremltools.converters.mil.frontend.tensorflow2.test.testing_utils import make_tf2_graph
  from coremltools.converters.mil.testing_utils import random_gen
label_desc: Use image node_id 2 for calling the TestControlFlowFromAutoGraph obj's underlying member method code with example usage: obj.test_if_unary_double_if_positive_else_square(compute_unit, backend) and returns: out