Skip to content

Commit

Permalink
test(polars): replace register with create_table
Browse files Browse the repository at this point in the history
  • Loading branch information
IndexSeek authored and jcrist committed Sep 11, 2024
1 parent 24be184 commit 76b44f3
Showing 1 changed file with 5 additions and 11 deletions.
16 changes: 5 additions & 11 deletions ibis/backends/polars/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
from typing import Any

import numpy as np
import polars as pl
import pytest

import ibis
Expand All @@ -23,16 +22,11 @@ def _load_data(self, **_: Any) -> None:
con = self.connection
for table_name in TEST_TABLES:
path = self.data_dir / "parquet" / f"{table_name}.parquet"
with pytest.warns(FutureWarning, match="v9.1"):
con.register(path, table_name=table_name)
# TODO: remove warnings and replace register when implementing 8858
with pytest.warns(FutureWarning, match="v9.1"):
con.register(array_types, table_name="array_types")
con.register(struct_types, table_name="struct")
con.register(win, table_name="win")

# TODO: remove when pyarrow inputs are supported
con._add_table("topk", pl.from_arrow(topk).lazy())
con.read_parquet(path, table_name=table_name)
con.create_table("array_types", array_types)
con.create_table("struct", struct_types)
con.create_table("win", win)
con.create_table("topk", topk)

@staticmethod
def connect(*, tmpdir, worker_id, **kw):
Expand Down

0 comments on commit 76b44f3

Please sign in to comment.