fix test

Author: liyixin

py-polars/tests/unit/operations/namespaces/list/test_list.py

@@ -386,12 +386,12 @@ def test_list_drop_nulls() -> None:
 def test_list_sample() -> None:
     s = pl.Series("values", [[1, 2, 3, None], [None, None], [1, 2], None])
 
-    expected_sample_n = pl.Series("values", [[3, 1], [None], [2], None])
+    expected_sample_n = pl.Series("values", [[None, 3], [None], [2], None])
     assert_series_equal(
         s.list.sample(n=pl.Series([2, 1, 1, 1]), seed=1), expected_sample_n
     )
 
-    expected_sample_frac = pl.Series("values", [[3, 1], [None], [1, 2], None])
+    expected_sample_frac = pl.Series("values", [[None, 3], [None], [1, 2], None])
     assert_series_equal(
         s.list.sample(fraction=pl.Series([0.5, 0.5, 1.0, 0.3]), seed=1),
         expected_sample_frac,
@@ -409,7 +409,10 @@ def test_list_sample() -> None:
         sample_frac=pl.col("values").list.sample(fraction=pl.col("frac"), seed=1),
     )
     expected_df = pl.DataFrame(
-        {"sample_n": [[3, 1], [None], [3, 4]], "sample_frac": [[3, 1], [None], [3, 4]]}
+        {
+            "sample_n": [[None, 3], [None], [3, 4]],
+            "sample_frac": [[None, 3], [None], [3, 4]],
+        }
     )
     assert_frame_equal(df, expected_df)
 

py-polars/tests/unit/operations/rolling/test_rolling.py

@@ -355,10 +355,10 @@ def test_rolling_extrema(dtype: PolarsDataType) -> None:
     # shuffled data triggers other kernels
     df = df.select([pl.all().shuffle(0)])
     expected = {
-        "col1": [None, None, 0, 0, 1, 2, 2],
-        "col2": [None, None, 0, 2, 1, 1, 1],
-        "col1_nulls": [None, None, None, None, None, 2, 2],
-        "col2_nulls": [None, None, None, None, None, 1, 1],
+        "col1": [None, None, 0, 0, 4, 1, 1],
+        "col2": [None, None, 1, 1, 0, 0, 0],
+        "col1_nulls": [None, None, None, None, 4, 4, 2],
+        "col2_nulls": [None, None, None, 1, 0, 0, 0],
     }
     result = df.select([pl.all().rolling_min(3)])
     assert result.to_dict(as_series=False) == {
@@ -367,10 +367,10 @@ def test_rolling_extrema(dtype: PolarsDataType) -> None:
 
     result = df.select([pl.all().rolling_max(3)])
     expected = {
-        "col1": [None, None, 6, 4, 5, 5, 5],
-        "col2": [None, None, 6, 6, 5, 4, 4],
-        "col1_nulls": [None, None, None, None, None, 5, 5],
-        "col2_nulls": [None, None, None, None, None, 4, 4],
+        "col1": [None, None, 5, 5, 6, 6, 6],
+        "col2": [None, None, 6, 6, 2, 5, 5],
+        "col1_nulls": [None, None, None, 5, 6, 6, 6],
+        "col2_nulls": [None, None, None, None, 2, 2, 4],
     }
     assert result.to_dict(as_series=False) == {
         k: pl.Series(v, dtype=dtype).to_list() for k, v in expected.items()
@@ -517,18 +517,18 @@ def test_rolling_group_by_extrema(dtype: PolarsDataType) -> None:
     expected = {
         "col1_list": pl.Series(
             [
-                [3],
-                [3, 4],
-                [3, 4, 5],
-                [4, 5, 6],
-                [5, 6, 2],
-                [6, 2, 1],
-                [2, 1, 0],
+                [4],
+                [4, 2],
+                [4, 2, 5],
+                [2, 5, 1],
+                [5, 1, 6],
+                [1, 6, 0],
+                [6, 0, 3],
             ],
             dtype=pl.List(dtype),
         ).to_list(),
-        "col1_min": pl.Series([3, 3, 3, 4, 2, 1, 0], dtype=dtype).to_list(),
-        "col1_max": pl.Series([3, 4, 5, 6, 6, 6, 2], dtype=dtype).to_list(),
+        "col1_min": pl.Series([4, 2, 2, 1, 1, 0, 0], dtype=dtype).to_list(),
+        "col1_max": pl.Series([4, 4, 5, 5, 6, 6, 6], dtype=dtype).to_list(),
     }
     assert result.to_dict(as_series=False) == expected
 

py-polars/tests/unit/operations/test_random.py

@@ -74,17 +74,17 @@ def test_sample_n_expr() -> None:
     )
 
     out_df = df.sample(pl.Series([3]), seed=0)
-    expected_df = pl.DataFrame({"group": [2, 2, 1], "val": [1, 1, 3]})
+    expected_df = pl.DataFrame({"group": [2, 1, 1], "val": [1, 2, 3]})
     assert_frame_equal(out_df, expected_df)
 
     agg_df = df.group_by("group", maintain_order=True).agg(
         pl.col("val").sample(pl.col("val").max(), seed=0)
     )
-    expected_df = pl.DataFrame({"group": [1, 2], "val": [[1, 2, 3], [1, 1]]})
+    expected_df = pl.DataFrame({"group": [1, 2], "val": [[1, 2, 3], [2, 1]]})
     assert_frame_equal(agg_df, expected_df)
 
     select_df = df.select(pl.col("val").sample(pl.col("val").max(), seed=0))
-    expected_df = pl.DataFrame({"val": [1, 1, 3]})
+    expected_df = pl.DataFrame({"val": [1, 2, 3]})
     assert_frame_equal(select_df, expected_df)
 
 
@@ -130,11 +130,11 @@ def test_shuffle_expr() -> None:
 
 def test_shuffle_series() -> None:
     a = pl.Series("a", [1, 2, 3])
-    out = a.shuffle(2)
-    expected = pl.Series("a", [2, 1, 3])
+    out = a.shuffle(1)
+    expected = pl.Series("a", [2, 3, 1])
     assert_series_equal(out, expected)
 
-    out = pl.select(pl.lit(a).shuffle(2)).to_series()
+    out = pl.select(pl.lit(a).shuffle(1)).to_series()
     assert_series_equal(out, expected)
 
 

py-polars/tests/unit/operations/test_rank.py

@@ -25,7 +25,7 @@ def test_rank_random_expr() -> None:
 def test_rank_random_series() -> None:
     s = pl.Series("a", [1, 2, 3, 2, 2, 3, 0])
     assert_series_equal(
-        s.rank("random", seed=1), pl.Series("a", [2, 4, 7, 3, 5, 6, 1], dtype=pl.UInt32)
+        s.rank("random", seed=1), pl.Series("a", [2, 5, 7, 3, 4, 6, 1], dtype=pl.UInt32)
     )