diff --git a/tests/expr_and_series/unary_test.py b/tests/expr_and_series/unary_test.py index dabab0c03..66afd22af 100644 --- a/tests/expr_and_series/unary_test.py +++ b/tests/expr_and_series/unary_test.py @@ -1,15 +1,11 @@ from typing import Any -import pytest - import narwhals.stable.v1 as nw from tests.utils import Constructor from tests.utils import compare_dicts -def test_unary(constructor: Constructor, request: pytest.FixtureRequest) -> None: - if "dask" in str(constructor): - request.applymarker(pytest.mark.xfail) +def test_unary(constructor: Constructor) -> None: data = {"a": [1, 3, 2], "b": [4, 4, 6], "z": [7.0, 8, 9]} result = ( nw.from_native(constructor(data)) @@ -20,7 +16,8 @@ def test_unary(constructor: Constructor, request: pytest.FixtureRequest) -> None z_min=nw.col("z").min(), z_max=nw.col("z").max(), ) - .select(nw.col("a_mean", "a_sum", "b_nunique", "z_min", "z_max").unique()) + .unique(["a_mean", "a_sum", "b_nunique", "z_min", "z_max"]) + .select(["a_mean", "a_sum", "b_nunique", "z_min", "z_max"]) ) expected = { "a_mean": [2],