def METHOD_NAME(self):
    layer_with_1_group = layers.GroupNormalization(
        groups=1, axis=-1, scale=False, center=False
    )
    layer_with_2_groups = layers.GroupNormalization(
        groups=2, axis=2, scale=False, center=False
    )
    inputs = np.array([[[-1.0, -1.0, 2.0, 2.0], [1.0, 1.0, 0, -2.0]]])
    expected_output_1_group = np.array(
        [[[-0.898, -0.898, 1.257, 1.257], [0.539, 0.539, -0.180, -1.616]]]
    )
    self.assertAllClose(
        layer_with_1_group(inputs),
        expected_output_1_group,
        atol=1e-3,
    )
    expected_output_2_groups = np.array(
        [[[-1.0, -1.0, 0.904, 0.904], [1.0, 1.0, -0.301, -1.507]]]
    )
    self.assertAllClose(
        layer_with_2_groups(inputs),
        expected_output_2_groups,
        atol=1e-3,
    )