code: def test_get_trial_dir(self, input_trial, expected_output):
      output = architecture_utils.DirectoryHandler.trial_dir(
          input_trial
      )
      self.assertEqual(output, expected_output)
  
imports: from absl.testing import parameterized
  from model_search import block_builder
  from model_search import hparam
  from model_search.architecture import architecture_utils
  from model_search.metadata import trial
  from model_search.proto import phoenix_spec_pb2
  from model_search.proto import transfer_learning_spec_pb2
  import numpy
  import tensorflow.compat.v2
label_desc: Use image node_id 5 for calling the ArchitectureUtilsTest obj's underlying member method code with example usage: obj.test_get_trial_dir(input_trial, expected_output) without return types