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