code: def testBuildKerasModel(self):
      built_model = model._build_keras_model(
          hidden_units=[1, 1], learning_rate=0.1
      )  # pylint: disable=protected-access
      self.assertEqual(len(built_model.layers), 10)
  
      built_model = model._build_keras_model(
          hidden_units=[1], learning_rate=0.1
      )  # pylint: disable=protected-access
      self.assertEqual(len(built_model.layers), 9)
  
imports: import tensorflow
  from tfx.experimental.templates.taxi.models.keras_model import model
label_desc: Use image node_id 1 for calling the ModelTest obj's underlying member method code with example usage: obj.testBuildKerasModel() without return types