def METHOD_NAME(self):
    xs = tf.ones((512, 512, 3))
    ys = tf.one_hot(tf.constant([1]), 2)
    inputs = {"images": xs, "labels": ys}
    layer = FourierMix()
    with self.assertRaisesRegexp(
        ValueError, "FourierMix received a single image to `call`"
    ):
        _ = layer(inputs)