Update tests

pola-rs · Jun 13, 2024 · cbe4091 · cbe4091
1 parent e4a5b6d
commit cbe4091
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/py-polars/tests/unit/dataframe/test_df.py b/py-polars/tests/unit/dataframe/test_df.py
@@ -1976,7 +1976,7 @@ def test_group_by_slice_expression_args() -> None:
 
     out = (
         df.group_by("groups", maintain_order=True)
-        .agg([pl.col("vals").slice(pl.len() * 0.1, (pl.len() // 5))])
+        .agg([pl.col("vals").slice((pl.len() * 0.1).cast(int), (pl.len() // 5))])
         .explode("vals")
     )
 

diff --git a/py-polars/tests/unit/expr/test_exprs.py b/py-polars/tests/unit/expr/test_exprs.py
@@ -678,7 +678,7 @@ def test_head() -> None:
     assert df.select(pl.col("a").head(10)).to_dict(as_series=False) == {
         "a": [1, 2, 3, 4, 5]
     }
-    assert df.select(pl.col("a").head(pl.len() / 2)).to_dict(as_series=False) == {
+    assert df.select(pl.col("a").head(pl.len() // 2)).to_dict(as_series=False) == {
         "a": [1, 2]
     }
 
@@ -690,7 +690,7 @@ def test_tail() -> None:
     assert df.select(pl.col("a").tail(10)).to_dict(as_series=False) == {
         "a": [1, 2, 3, 4, 5]
     }
-    assert df.select(pl.col("a").tail(pl.len() / 2)).to_dict(as_series=False) == {
+    assert df.select(pl.col("a").tail(pl.len() // 2)).to_dict(as_series=False) == {
         "a": [4, 5]
     }