NO-SNOW: Apply test session parameters to doctest sessions (#4209)

Author: sfc-gh-bkogan

src/conftest.py

@@ -22,6 +22,8 @@
 LOCAL_TESTING_MODE = False
 
 sys.path.append("tests/")
+from integ.session_parameters import set_up_test_session_parameters  # noqa: E402
+
 # get the absolute path of rsa private key files
 params_file = os.path.abspath("tests/parameters.py")
 with open("tests/parameters.py", encoding="utf-8") as f:
@@ -67,6 +69,7 @@ def add_snowpark_session(doctest_namespace, pytestconfig):
         session.sql_simplifier_enabled = (
             os.environ.get("USE_SQL_SIMPLIFIER") == "1" or LOCAL_TESTING_MODE
         )
+        set_up_test_session_parameters(session, LOCAL_TESTING_MODE)
         if RUNNING_ON_GH:
             session.sql(f"CREATE SCHEMA IF NOT EXISTS {TEST_SCHEMA}").collect()
             # This is needed for test_get_schema_database_works_after_use_role in test_session_suite

src/snowflake/snowpark/session.py

@@ -1679,10 +1679,10 @@ def add_packages(
             >>> import pandas
             >>> import dateutil
             >>> # add numpy with the latest version on Snowflake Anaconda
-            >>> # and pandas with the version "2.1.*"
+            >>> # and pandas with the version "2.3.*"
             >>> # and dateutil with the local version in your environment
             >>> session.custom_package_usage_config = {"enabled": True}  # This is added because latest dateutil is not in snowflake yet
-            >>> session.add_packages("numpy", "pandas==2.2.*", dateutil)
+            >>> session.add_packages("numpy", "pandas==2.3.*", dateutil)
             >>> @udf
             ... def get_package_name_udf() -> list:
             ...     return [numpy.__name__, pandas.__name__, dateutil.__name__]
@@ -1739,7 +1739,7 @@ def remove_package(
             >>> session.clear_packages()
             >>> len(session.get_packages())
             0
-            >>> session.add_packages("numpy", "pandas==2.2.*")
+            >>> session.add_packages("numpy", "pandas==2.3.*")
             >>> len(session.get_packages())
             2
             >>> session.remove_package("numpy")