code: def _format_test_variant(alleles, call_infos):
      variant = test_utils.make_variant(
          chrom="20", start=0, alleles=alleles
      )
      for i, call_info in enumerate(call_infos):
          call = variant.calls.add(call_set_name="sample" + str(i))
          for key, value in call_info.items():
              if not isinstance(value, (list, tuple)):
                  value = [value]
              call.info[key].values.extend(
                  [struct_pb2.Value(int_value=v) for v in value]
              )
      return variant
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import absltest
  from absl.testing import parameterized
  from etils import epath
  from third_party.nucleus.io import vcf
  from third_party.nucleus.protos import reference_pb2
  from third_party.nucleus.protos import struct_pb2
  from third_party.nucleus.protos import variants_pb2
  from third_party.nucleus.testing import test_utils
  from third_party.nucleus.util import ranges
label_desc: Use image node_id 2 for calling a global function with example usage: _format_test_variant(alleles, call_infos) and returns: variant