test: Add unit test for get_python_version

Author: jialuoo

tests/unit/functions/test_remote_function_utils.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import inspect
+import sys
 from unittest.mock import patch
 
 import bigframes_vendored.constants as constants
@@ -227,6 +228,28 @@ def test_get_updated_package_requirements_with_existing_cloudpickle():
     assert result == expected
 
 
+# Dynamically generate expected python versions for the test
+_major = sys.version_info.major
+_minor = sys.version_info.minor
+_compat_version = f"python{_major}{_minor}"
+_standard_version = f"python-{_major}.{_minor}"
+
+
+@pytest.mark.parametrize(
+    "is_compat, expected_version",
+    [
+        (True, _compat_version),
+        (False, _standard_version),
+    ],
+)
+def test_get_python_version(is_compat, expected_version):
+    """
+    Tests the python version string formatting for both standard and compat modes.
+    """
+    result = _utils.get_python_version(is_compat=is_compat)
+    assert result == expected_version
+
+
 def test_package_existed_helper():
     """Tests the _package_existed helper function directly."""
     reqs = ["pandas==1.0", "numpy", "scikit-learn>=1.2.0"]