update tests

snowflakedb · Dec 17, 2024 · 32b7795 · 32b7795
1 parent 5353f51
commit 32b7795
Show file tree

Hide file tree

Showing 3 changed files with 4,760 additions and 10,104 deletions.
diff --git a/src/snowflake/snowpark/functions.py b/src/snowflake/snowpark/functions.py
@@ -5258,15 +5258,15 @@ def sequence(
     start_col = _to_col_if_str(start, "sequence")
     stop_col = _to_col_if_str(stop, "sequence")
     if step is None:
-        step = iff(builtin("sign", _emit_ast=False)(stop_col - start_col) > 0, 1, -1)
+        step = iff(builtin("sign")(stop_col - start_col) > 0, 1, -1)
         ans = builtin("array_generate_range", _emit_ast=False)(
             start_col, stop_col + step, step
         )
         ans._ast = ast
         return ans
 
     step_col = _to_col_if_str(step, "sequence")
-    step_sign = iff(builtin("sign", _emit_ast=False)(step_col) > 0, 1, -1)
+    step_sign = iff(builtin("sign")(step_col) > 0, 1, -1)
     ans = builtin("array_generate_range", _emit_ast=False)(
         start_col, stop_col + step_sign, step_col
     )

diff --git a/tests/ast/data/Dataframe.to_snowpark_pandas.test b/tests/ast/data/Dataframe.to_snowpark_pandas.test
@@ -35,7 +35,10 @@ body {
           }
         }
         src {
+          end_column: 41
+          end_line: 25
           file: "SRC_POSITION_TEST_MODE"
+          start_column: 13
           start_line: 25
         }
         variant {
@@ -64,7 +67,10 @@ body {
           }
         }
         src {
+          end_column: 52
+          end_line: 27
           file: "SRC_POSITION_TEST_MODE"
+          start_column: 29
           start_line: 27
         }
       }
@@ -93,7 +99,10 @@ body {
           list: "A"
         }
         src {
+          end_column: 65
+          end_line: 29
           file: "SRC_POSITION_TEST_MODE"
+          start_column: 29
           start_line: 29
         }
       }
@@ -123,7 +132,10 @@ body {
           }
         }
         src {
+          end_column: 70
+          end_line: 31
           file: "SRC_POSITION_TEST_MODE"
+          start_column: 29
           start_line: 31
         }
       }
@@ -156,7 +168,10 @@ body {
           list: "A"
         }
         src {
+          end_column: 87
+          end_line: 33
           file: "SRC_POSITION_TEST_MODE"
+          start_column: 29
           start_line: 33
         }
       }
@@ -183,5 +198,5 @@ client_language {
 }
 client_version {
   major: 1
-  minor: 25
+  minor: 26
 }