code: def test_get_architecture_size(self, architecture):
      # Force graph mode
      with tf.compat.v1.Graph().as_default():
          with self.test_session(graph=tf.Graph()):
              tower_name = "tower"
              if architecture is not None:
                  architecture_utils.set_architecture(
                      architecture, tower_name=tower_name
                  )
              size = architecture_utils.get_architecture_size(
                  tower_name=tower_name
              )
              if architecture is not None:
                  self.assertEqual(size, architecture.size)
              else:
                  self.assertIsNone(size)
  
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 9 for calling the ArchitectureUtilsTest obj's underlying member method code with example usage: obj.test_get_architecture_size(architecture) without return types