code: def test_custom_column(self):
      # This column is not in any allowlist but should succeed because
      # it inherits from V2 CategoricalColumn.
      categorical_column = fc_lib.categorical_column_with_identity(
          key="aaa", num_buckets=10
      )
      embedding_dimension = 2
      embedding_column = tpu_fc.embedding_column(
          categorical_column, dimension=embedding_dimension
      )
      self.assertIs(
          categorical_column, embedding_column.categorical_column
      )
      self.assertEqual(embedding_dimension, embedding_column.dimension)
      self.assertEqual("mean", embedding_column.combiner)
      self.assertEqual("aaa_embedding", embedding_column.name)
      self.assertEqual(
          "aaa_embedding", embedding_column._var_scope_name
      )
      self.assertEqual(
          (embedding_dimension,), embedding_column._variable_shape
      )
      self.assertEqual(
          {"aaa": parsing_ops.VarLenFeature(dtypes.int64)},
          embedding_column._parse_example_spec,
      )
  
imports: import numpy
  from tensorflow.python.client import session
  from tensorflow.python.feature_column import feature_column
  from tensorflow.python.feature_column import feature_column_lib
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import sparse_tensor
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import lookup_ops
  from tensorflow.python.ops import parsing_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
  from tensorflow.python.tpu import feature_column
label_desc: Use image node_id 3 for calling the EmbeddingColumnTest obj's underlying member method code with example usage: obj.test_custom_column() without return types