def test_linear_model_two_input_values(self):
    """Tests linear_model() for input with shape=[2]."""
    price = fc._numeric_column('price', shape=[2])
    bucketized_price = fc._bucketized_column(price, boundaries=[0, 2, 4, 6])
    with ops.Graph().as_default():