From 76b44f357ff729a324192ea811443ee57b50205d Mon Sep 17 00:00:00 2001 From: Tyler White <50381805+IndexSeek@users.noreply.github.com> Date: Wed, 11 Sep 2024 01:18:24 +0000 Subject: [PATCH] test(polars): replace register with create_table --- ibis/backends/polars/tests/conftest.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/ibis/backends/polars/tests/conftest.py b/ibis/backends/polars/tests/conftest.py index 1c0db81be687..8aec5127e305 100644 --- a/ibis/backends/polars/tests/conftest.py +++ b/ibis/backends/polars/tests/conftest.py @@ -3,7 +3,6 @@ from typing import Any import numpy as np -import polars as pl import pytest import ibis @@ -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):