import pytest

from tests_backend.conftest import TEST_LABEL_ID


@pytest.fixture
def results_as_df(snowflake_client):
    """Get the results of a query as a DataFrame."""

    async def _results_as_df(sql_func):
        sql = sql_func(TEST_LABEL_ID)
        return await snowflake_client.afetch_all(sql, as_df=True)

    return _results_as_df