def METHOD_NAME(self):
    test_utils.layer_test(
        keras.layers.Dropout, kwargs={"rate": 0.5}, input_shape=(3, 2)
    )
    test_utils.layer_test(
        keras.layers.Dropout,
        kwargs={"rate": 0.5, "noise_shape": [3, 1]},
        input_shape=(3, 2),
    )