Skip to content

Commit

Permalink
Update test references
Browse files Browse the repository at this point in the history
  • Loading branch information
stinodego committed Feb 29, 2024
1 parent bd8ae08 commit 8530a56
Show file tree
Hide file tree
Showing 14 changed files with 16 additions and 16 deletions.
6 changes: 3 additions & 3 deletions .github/scripts/test_bytecode_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from typing import Any, Callable

import pytest
from polars.utils.udfs import BytecodeParser
from polars._utils.udfs import BytecodeParser
from tests.unit.operations.map.test_inefficient_map_warning import (
MY_DICT,
NOOP_TEST_CASES,
Expand All @@ -44,7 +44,7 @@ def test_bytecode_parser_expression_in_ipython(
col: str, func: Callable[[Any], Any], expected: str
) -> None:
script = (
"from polars.utils.udfs import BytecodeParser; "
"from polars._utils.udfs import BytecodeParser; "
"import datetime as dt; "
"from datetime import datetime; "
"import numpy as np; "
Expand Down Expand Up @@ -73,7 +73,7 @@ def test_bytecode_parser_expression_noop(func: str) -> None:
)
def test_bytecode_parser_expression_noop_in_ipython(func: str) -> None:
script = (
"from polars.utils.udfs import BytecodeParser; "
"from polars._utils.udfs import BytecodeParser; "
f"MY_DICT = {MY_DICT};"
f'parser = BytecodeParser({func}, map_target="expr");'
f'print(not parser.can_attempt_rewrite() or not parser.to_expression("x"));'
Expand Down
2 changes: 1 addition & 1 deletion docs/user-guide/expressions/plugins.md
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ expression in batches. Whereas for other operations this would not be allowed, t
# expression_lib/__init__.py
import polars as pl
from polars.type_aliases import IntoExpr
from polars.utils.udfs import _get_shared_lib_location
from polars._utils.udfs import _get_shared_lib_location

from expression_lib.utils import parse_into_expr

Expand Down
2 changes: 1 addition & 1 deletion py-polars/polars/dataframe/frame.py
Original file line number Diff line number Diff line change
Expand Up @@ -6498,7 +6498,7 @@ def map_rows(
>>> df.select(pl.col("foo") * 2 + pl.col("bar")) # doctest: +IGNORE_RESULT
"""
# TODO: Enable warning for inefficient map
# from polars.utils.udfs import warn_on_inefficient_map
# from polars._utils.udfs import warn_on_inefficient_map
# warn_on_inefficient_map(function, columns=self.columns, map_target="frame)

out, is_df = self._df.map_rows(function, return_dtype, inference_size)
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/constructors/test_constructors.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@

from polars.datatypes import PolarsDataType
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


# -----------------------------------------------------------------------------------
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/dataframe/test_df.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@

from polars.type_aliases import JoinStrategy, UniqueKeepStrategy
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


def test_version() -> None:
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/datatypes/test_temporal.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@

from polars.type_aliases import Ambiguous, PolarsTemporalType, TimeUnit
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


def test_fill_null() -> None:
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/expr/test_exprs.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
if TYPE_CHECKING:
from zoneinfo import ZoneInfo
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


def test_arg_true() -> None:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@

from polars.type_aliases import TimeUnit
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


def test_date_datetime() -> None:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
from polars.datatypes import PolarsDataType
from polars.type_aliases import ClosedInterval, TimeUnit
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


def test_datetime_range() -> None:
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/namespaces/test_datetime.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@

from polars.type_aliases import TemporalLiteral, TimeUnit
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


@pytest.fixture()
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/namespaces/test_strptime.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@

from polars.type_aliases import PolarsTemporalType, TimeUnit
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


def test_str_strptime() -> None:
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/operations/test_group_by_dynamic.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

from polars.type_aliases import Label, StartBy
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


@pytest.mark.parametrize(
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/series/test_series.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@

from polars.type_aliases import EpochTimeUnit, PolarsDataType, TimeUnit
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


def test_cum_agg() -> None:
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/unit/utils/test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@

from polars.type_aliases import TimeUnit
else:
from polars.utils.convert import string_to_zoneinfo as ZoneInfo
from polars._utils.convert import string_to_zoneinfo as ZoneInfo


@pytest.mark.parametrize(
Expand Down

0 comments on commit 8530a56

Please sign in to comment.