test: fix test failure from merge (#2007)

This commit is contained in:
Will Jones
2025-01-09 11:27:24 -08:00
committed by GitHub
parent d3ea75cc2b
commit 6eacae18c4
3 changed files with 4 additions and 4 deletions

View File

@@ -449,10 +449,10 @@ async def test_query_to_pandas_flatten_async(table_struct_async: AsyncTable):
@pytest.mark.asyncio
async def test_query_to_polars_async(table_async: AsyncTable):
df = await table_async.query().to_polars()
assert df.shape == (2, 4)
assert df.shape == (2, 5)
df = await table_async.query().where("id < 0").to_polars()
assert df.shape == (0, 4)
assert df.shape == (0, 5)
@pytest.mark.asyncio