def METHOD_NAME(self):
    layer = layers.Resizing(8, 9)
    input_data = np.random.random((2, 10, 12, 3))
    ds = tf_data.Dataset.from_tensor_slices(input_data).batch(2).map(layer)
    for output in ds.take(1):
        output = output.numpy()
    self.assertEqual(list(output.shape), [2, 8, 9, 3])