code: def test_model_build(base_layer_fn, input_shape, data_init):
      inputs = tf.keras.layers.Input(shape=input_shape)
      base_layer = base_layer_fn()
      wt_layer = wrappers.WeightNormalization(base_layer, data_init)
      model = tf.keras.models.Sequential(layers=[inputs, wt_layer])
      model.build()
  
file_path: addons/tensorflow_addons/layers/tests/wrappers_test.py
imports: import os
  import tempfile
  import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.layers import wrappers
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 10 for calling a global function with example usage: test_model_build(base_layer_fn, input_shape, data_init) without return types