From 5d831092bf0016a1786beb02aafbdd88d8430bcc Mon Sep 17 00:00:00 2001 From: ritchie Date: Mon, 11 Nov 2024 16:26:08 +0100 Subject: [PATCH] lint --- py-polars/tests/unit/operations/test_top_k.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/py-polars/tests/unit/operations/test_top_k.py b/py-polars/tests/unit/operations/test_top_k.py index 2eb10c8785f0..e0aa011d342a 100644 --- a/py-polars/tests/unit/operations/test_top_k.py +++ b/py-polars/tests/unit/operations/test_top_k.py @@ -409,21 +409,21 @@ def test_top_k_df() -> None: assert df.bottom_k(3, by="a").collect()["a"].to_list() == expected df = pl.LazyFrame({"a": [1, None, None, 4, 5]}) - expected = [5, 4, 1, None] + expected2 = [5, 4, 1, None] assert ( df.sort("a", descending=True, nulls_last=True).limit(4).collect()["a"].to_list() == expected ) - assert df.top_k(4, by="a").collect()["a"].to_list() == expected - expected = [1, 4, 5, None] + assert df.top_k(4, by="a").collect()["a"].to_list() == expected2 + expected2 = [1, 4, 5, None] assert ( df.sort("a", descending=False, nulls_last=True) .limit(4) .collect()["a"] .to_list() - == expected + == expected2 ) - assert df.bottom_k(4, by="a").collect()["a"].to_list() == expected + assert df.bottom_k(4, by="a").collect()["a"].to_list() == expected2 assert df.sort("a", descending=False, nulls_last=False).limit(4).collect()[ "a"