def METHOD_NAME(data_init, input_data, expected_output):
    layer = tf.keras.layers.Dense(
        input_data.shape[-1],
        activation=None,
        kernel_initializer="identity",
        bias_initializer="zeros",
    )
    test_utils.layer_test(
        wrappers.WeightNormalization,
        kwargs={"layer": layer, "data_init": data_init},
        input_data=input_data,
        expected_output=expected_output,
    )