NO-SNOW: fix flaky tests (#4061)

Author: sfc-gh-mayliu

src/snowflake/snowpark/dataframe.py

@@ -385,7 +385,7 @@ class DataFrame:
         >>> with tempfile.NamedTemporaryFile(mode="w+t") as t:
         ...     t.writelines(["id1, Product A", "\\n" "id2, Product B"])
         ...     t.flush()
-        ...     create_stage_result = session.sql("create temp stage test_stage").collect()
+        ...     create_stage_result = session.sql("create or replace temp stage test_stage").collect()
         ...     put_result = session.file.put(t.name, "@test_stage/test_dir")
 
     Example 1
@@ -418,7 +418,7 @@ class DataFrame:
             |2        |two      |
             ---------------------
             <BLANKLINE>
-            >>> session.range(1, 10, 2).to_df("col1").show()
+            >>> session.range(1, 10, 2).to_df("col1").sort("col1").show()
             ----------
             |"COL1"  |
             ----------
@@ -4641,7 +4641,7 @@ def write(self) -> DataFrameWriter:
             |3    |4    |
             -------------
             <BLANKLINE>
-            >>> stage_created_result = session.sql("create temp stage if not exists test_stage").collect()
+            >>> stage_created_result = session.sql("create or replace temp stage test_stage").collect()
             >>> df.write.copy_into_location("@test_stage/copied_from_dataframe")  # default CSV
             [Row(rows_unloaded=2, input_bytes=8, output_bytes=28)]
         """
@@ -4689,7 +4689,7 @@ def copy_into_table(
             >>> with tempfile.NamedTemporaryFile(mode="w+t") as t:
             ...     t.writelines(["id1, Product A", "\\n" "id2, Product B"])
             ...     t.flush()
-            ...     create_stage_result = session.sql("create temp stage if not exists test_stage").collect()
+            ...     create_stage_result = session.sql("create or replace temp stage test_stage").collect()
             ...     put_result = session.file.put(t.name, "@test_stage/copy_into_table_dir", overwrite=True)
             >>> # user_schema is used to read from CSV files. For other files it's not needed.
             >>> from snowflake.snowpark.types import StringType, StructField, StringType

src/snowflake/snowpark/session.py

@@ -2856,7 +2856,7 @@ def generator(
         Example 1
             >>> from snowflake.snowpark.functions import seq1, seq8, uniform
             >>> df = session.generator(seq1(1).as_("sequence one"), uniform(1, 10, 2).as_("uniform"), rowcount=3)
-            >>> df.show()
+            >>> df.sort("sequence one").show()
             ------------------------------
             |"sequence one"  |"UNIFORM"  |
             ------------------------------

tests/integ/modin/index/test_reindex.py

@@ -170,7 +170,7 @@ def test_non_overlapping_different_types_index_negative_SNOW_1622502():
 
     with pytest.raises(
         SnowparkSQLException,
-        match='Failed to cast variant value "A" to TIMESTAMP_NTZ',
+        match='Failed to cast variant value "[ABC]" to TIMESTAMP_NTZ',
     ):
         date_index.reindex(list("ABC"))