chore: import upstream snapshot with attribution

release/ray_release/tests/test_template.py

@@ -0,0 +1,56 @@
+import sys
+
+import pytest
+
+from ray_release.exception import ReleaseTestConfigError
+from ray_release.template import bazel_runfile, get_working_dir
+from ray_release.test import Test
+
+
+def test_get_working_dir_with_path_from_root():
+    test_with_path_from_root = Test(
+        {
+            "name": "test",
+            "working_dir": "//ray_testing/ray_release/tests",
+        }
+    )
+    assert (
+        get_working_dir(test_with_path_from_root, None, "/tmp/bazel_workspace")
+        == "/tmp/bazel_workspace/ray_testing/ray_release/tests"
+    )
+    assert get_working_dir(test_with_path_from_root, None, None) == bazel_runfile(
+        "ray_testing/ray_release/tests"
+    )
+
+
+def test_get_working_dir_with_relative_path():
+    test_with_relative_path = Test(
+        {
+            "name": "test",
+            "working_dir": "ray_release/tests",
+        }
+    )
+    assert (
+        get_working_dir(test_with_relative_path, None, "/tmp/bazel_workspace")
+        == "/tmp/bazel_workspace/release/ray_release/tests"
+    )
+    assert get_working_dir(test_with_relative_path, None, None) == bazel_runfile(
+        "release/ray_release/tests"
+    )
+
+
+def test_get_working_dir_fail():
+    test_with_path_from_root = Test(
+        {
+            "name": "test",
+            "working_dir": "//ray_testing/ray_release/tests",
+        }
+    )
+    with pytest.raises(ReleaseTestConfigError):
+        get_working_dir(
+            test_with_path_from_root, "/tmp/test_definition_root", "tmp/bazel_workspace"
+        )
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-v", __file__]))