176d04bd5187c84f2cf9b89f84bbfe19d61c4c3d,tests/autokeras/adapters/input_adapter_test.py,,test_time_series_input_less_col_name,#,236

Before Change

def test_time_series_input_less_col_name():
    (x, _), _1 = utils.dataframe_numpy()
    with pytest.raises(ValueError) as info:
        adapter = input_adapter.TimeseriesInputAdapter(
            lookback=2, column_names=utils.LESS_COLUMN_NAMES_FROM_CSV
        )

After Change

        adapter = input_adapter.TimeseriesInputAdapter(
            lookback=2, column_names=utils.COLUMN_NAMES[:-2]
        )
        adapter.transform(pd.read_csv(utils.TRAIN_CSV_PATH))
    assert "Expect column_names to have length" in str(info.value)

Instances