fix test

pola-rs · Oct 25, 2023 · b913b3c · b913b3c
1 parent d051153
commit b913b3c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/crates/polars/tests/it/lazy/expressions/arity.rs b/crates/polars/tests/it/lazy/expressions/arity.rs
@@ -260,7 +260,7 @@ fn test_binary_over_3930() -> PolarsResult<()> {
     ]?;
 
     let ss = col("score").pow(2);
-    let mdiff = (ss.clone().shift(-1) - ss.shift(1)) / lit(2);
+    let mdiff = (ss.clone().shift(lit(-1)) - ss.shift(lit(1))) / lit(2);
     let out = df.lazy().select([mdiff.over([col("class")])]).collect()?;
 
     let out = out.column("score")?;

diff --git a/crates/polars/tests/it/lazy/projection_queries.rs b/crates/polars/tests/it/lazy/projection_queries.rs
@@ -136,7 +136,7 @@ fn test_projection_5086() -> PolarsResult<()> {
         .select([
             col("a"),
             col("b")
-                .xor(col("b").shift(1).over([col("a")]))
+                .xor(col("b").shift(lit(1)).over([col("a")]))
                 .fill_null(lit(true))
                 .alias("keep"),
         ])