chore: import upstream snapshot with attribution

python/ray/tests/test_error_ray_not_initialized.py

@@ -0,0 +1,50 @@
+import sys
+
+import pytest
+
+import ray
+
+
+def test_errors_before_initializing_ray(set_enable_auto_connect):
+    @ray.remote
+    def f():
+        pass
+
+    @ray.remote
+    class Foo:
+        pass
+
+    api_methods = [
+        f.remote,
+        Foo.remote,
+        lambda: ray.cancel(None),  # Not valid API usage.
+        lambda: ray.get([]),
+        lambda: ray.get_actor("name"),
+        ray.get_gpu_ids,
+        lambda: ray.kill(None),  # Not valid API usage.
+        ray.nodes,
+        lambda: ray.put(1),
+        lambda: ray.wait([]),
+    ]
+
+    def test_exceptions_raised():
+        for api_method in api_methods:
+            print(api_method)
+            with pytest.raises(
+                ray.exceptions.RaySystemError, match="Ray has not been started yet."
+            ):
+                api_method()
+
+    test_exceptions_raised()
+
+    # Make sure that the exceptions are still raised after Ray has been
+    # started and shutdown.
+    ray.init(num_cpus=0)
+    ray.shutdown()
+
+    test_exceptions_raised()
+
+
+if __name__ == "__main__":
+
+    sys.exit(pytest.main(["-sv", __file__]))