diff --git a/py-polars/tests/unit/operations/test_top_k.py b/py-polars/tests/unit/operations/test_top_k.py index e0aa011d342a..866ef88e6e10 100644 --- a/py-polars/tests/unit/operations/test_top_k.py +++ b/py-polars/tests/unit/operations/test_top_k.py @@ -412,7 +412,7 @@ def test_top_k_df() -> None: expected2 = [5, 4, 1, None] assert ( df.sort("a", descending=True, nulls_last=True).limit(4).collect()["a"].to_list() - == expected + == expected2 ) assert df.top_k(4, by="a").collect()["a"].to_list() == expected2 expected2 = [1, 4, 5, None]