def METHOD_NAME(self):
    df = (
        FreightPricingSearch()
        .search(routes=["TD3C"])
        .to_df(
            columns=[
                "short_code",
                "rate",
                "rate_unit",
                "prediction.outlook_1d.prediction",
            ]
        )
        .head(2)
    )
    assert len(df) == 2
    assert list(df.columns) == [
        "short_code",
        "rate",
        "rate_unit",
        "prediction.outlook_1d.prediction",
    ]