code: def test_get_trial_mode(
      self,
      distillation_type,
      ensembling_type,
      distillation_pool,
      n,
      expected,
  ):
      trial_id = 50
  
      distillation_spec = distillation_spec_pb2.DistillationSpec()
      distillation_spec.distillation_type = distillation_type
      distillation_spec.minimal_pool_size = distillation_pool
  
      ensembling_spec = self._create_ensemble_spec(ensembling_type, n)
  
      actual = trial_utils.get_trial_mode(
          ensembling_spec, distillation_spec, trial_id
      )
      self.assertEqual(actual, expected)
  
imports: import functools
  import os
  from absl.testing import absltest
  from absl.testing import parameterized
  import mock
  from model_search import hparam
  from model_search.architecture import architecture_utils
  from model_search.generators import trial_utils
  from model_search.proto import distillation_spec_pb2
  from model_search.proto import ensembling_spec_pb2
  from model_search.proto import phoenix_spec_pb2
  import tensorflow.compat.v2
  from google.protobuf import text_format
label_desc: Use image node_id 2 for calling the TrialUtilsTest obj's underlying member method code with example usage: obj.test_get_trial_mode(distillation_type, ensembling_type, distillation_pool, n, expected) without return types