def METHOD_NAME(self):
    dummy_inputs = self.rng.uniform(shape=(32, 224, 224, 3))
    layer = RandomApply(rate=0.0, layer=ZeroOut())
    outputs = layer(dummy_inputs)
    self.assertAllClose(outputs, dummy_inputs)