def METHOD_NAME(self):
    int_values = keras.Input(shape=(2,), dtype=tf.int32)
    float_values = tf.cast(int_values, tf.float32)
    model = keras.Model(int_values, float_values)
    model.compile(loss="mse")
    input_data = np.array([[1, 2], [3, 4]], dtype=np.int32)
    expected = [[1.0, 2.0], [3.0, 4.0]]
    output = model.predict(input_data)
    self.assertAllClose(expected, output)