chore: import upstream snapshot with attribution

python/ray/tests/test_typing.py

@@ -0,0 +1,60 @@
+import os
+import shutil
+import subprocess
+import sys
+import tempfile
+
+import mypy.api as mypy_api
+import pyright
+import pytest
+
+# Paths are relative to the directory where Bazel is run in the CI
+TYPING_GOOD_PATH = "python/ray/tests/typing_files/check_typing_good.py"
+TYPING_BAD_PATH = "python/ray/tests/typing_files/check_typing_bad.py"
+TYPING_ACTOR_ASYNC_PATH = "python/ray/tests/typing_files/check_typing_actor_async.py"
+
+
+def test_typing_good():
+    typing_good_tmp_path = create_tmp_copy(TYPING_GOOD_PATH)
+    out, msg, status_code = mypy_api.run([typing_good_tmp_path])
+    assert status_code == 0, out
+
+
+def test_typing_bad():
+    typing_bad_tmp_path = create_tmp_copy(TYPING_BAD_PATH)
+    _, msg, status_code = mypy_api.run([typing_bad_tmp_path])
+    assert status_code == 1, msg
+
+
+def test_typing_actor_async():
+    typing_actor_async_tmp_path = create_tmp_copy(TYPING_ACTOR_ASYNC_PATH)
+    result = pyright.run(
+        typing_actor_async_tmp_path,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+        text=True,
+        cwd=tempfile.gettempdir(),
+    )
+    assert (
+        result.returncode == 0
+    ), f"Pyright check failed. stdout:\n{result.stdout}\n\nstderr:\n{result.stderr}"
+
+
+def create_tmp_copy(file_path: str) -> str:
+    """Copies file at file_path to a temporary file and returns the path."""
+
+    base_file_name = os.path.basename(file_path)
+    tmp_dir = tempfile.gettempdir()
+    tmp_path = os.path.join(tmp_dir, base_file_name)
+    shutil.copy(file_path, tmp_path)
+    print(f"Copied file at {file_path} to {tmp_path}")
+
+    return tmp_path
+
+
+if __name__ == "__main__":
+    # Make subprocess happy in bazel.
+    os.environ["LC_ALL"] = "en_US.UTF-8"
+    os.environ["LANG"] = "en_US.UTF-8"
+
+    sys.exit(pytest.main(["-sv", __file__]))