improve and fix test

Author: sfc-gh-aling

src/snowflake/snowpark/session.py

@@ -1730,8 +1730,10 @@ def add_requirements(
             >>> from snowflake.snowpark.functions import udf
             >>> import numpy
             >>> import pandas
+            >>> import sys
             >>> # test_requirements.txt contains "numpy" and "pandas"
-            >>> session.add_requirements("tests/resources/test_requirements.txt")
+            >>> file = "test_requirements.txt" if sys.version_info < (3, 13) else "test_requirements_py313.txt"
+            >>> session.add_requirements(f"tests/resources/{file}")
             >>> @udf
             ... def get_package_name_udf() -> list:
             ...     return [numpy.__name__, pandas.__name__]

…equirements_greater_than_python_3_12.txt …ts/resources/test_requirements_py313.txttests/resources/test_requirements_greater_than_python_3_12.txt renamed to tests/resources/test_requirements_py313.txt tests/resources/test_requirements_greater_than_python_3_12.txt renamed to tests/resources/test_requirements_py313.txt

@@ -350,7 +350,7 @@ def check_zip_files_and_close_stream(input_stream, expected_files):
                 "resources/test_all_data_types.parquet",
                 "resources/test_file_with_special_characters.parquet",
                 "resources/test_requirements.txt",
-                "resources/test_requirements_greater_than_python_3_12.txt",
+                "resources/test_requirements_py313.txt",
                 "resources/test_requirements_unsupported.txt",
                 "resources/test_concat_file1.csv",
                 "resources/test_concat_file2.csv",

tests/unit/scala/test_utils_suite.py

@@ -1633,10 +1633,12 @@ def test_pandas_udf_py_file(self):
 
     @property
     def test_requirements_file(self):
+        # TODO: SNOW-2389419, snowpark does not support multiple line requirements for a single package
+        # ideally the req file could be like:
+        # numpy==1.26.3; python_version<"3.13"
+        # numpy==2.3.1; python_version>="3.13"
         if sys.version_info >= (3, 13):
-            return os.path.join(
-                self.resources_path, "test_requirements_greater_than_python_3_12.txt"
-            )
+            return os.path.join(self.resources_path, "test_requirements_py313.txt")
         return os.path.join(self.resources_path, "test_requirements.txt")
 
     @property