Use pytest not unittest (ray-project#16265)

* .

* done

* done

* .

* .

Co-authored-by: Alex Wu <alex@anyscale.com>

Author: Alex Wu

python/ray/tests/test_runtime_env.py

@@ -1,7 +1,6 @@
 import os
 import pytest
 import sys
-import unittest
 import random
 import tempfile
 import requests
@@ -123,7 +122,7 @@ def start_client_server(cluster, client_mode):
     return ("localhost:10003", {"USE_RAY_CLIENT": "1"}, PKG_DIR)
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 def test_travel():
     import uuid
     with tempfile.TemporaryDirectory() as tmp_dir:
@@ -195,7 +194,7 @@ def handler(path):
 """
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_empty_working_dir(ray_start_cluster_head, client_mode):
     cluster = ray_start_cluster_head
@@ -213,7 +212,7 @@ def test_empty_working_dir(ray_start_cluster_head, client_mode):
         assert out != "ERROR"
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_invalid_working_dir(ray_start_cluster_head, working_dir, client_mode):
     cluster = ray_start_cluster_head
@@ -249,7 +248,7 @@ def test_invalid_working_dir(ray_start_cluster_head, working_dir, client_mode):
     assert out == "ValueError"
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_single_node(ray_start_cluster_head, working_dir, client_mode):
     cluster = ray_start_cluster_head
@@ -265,7 +264,7 @@ def test_single_node(ray_start_cluster_head, working_dir, client_mode):
     assert len(kv._internal_kv_list("gcs://")) == 0
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_two_node(two_node_cluster, working_dir, client_mode):
     cluster, _ = two_node_cluster
@@ -281,7 +280,7 @@ def test_two_node(two_node_cluster, working_dir, client_mode):
     assert len(kv._internal_kv_list("gcs://")) == 0
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_two_node_module(two_node_cluster, working_dir, client_mode):
     cluster, _ = two_node_cluster
@@ -296,7 +295,7 @@ def test_two_node_module(two_node_cluster, working_dir, client_mode):
     assert len(list(Path(PKG_DIR).iterdir())) == 1
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_two_node_local_file(two_node_cluster, working_dir, client_mode):
     with open(os.path.join(working_dir, "test_file"), "w") as f:
@@ -317,7 +316,7 @@ def test_two_node_local_file(two_node_cluster, working_dir, client_mode):
     assert len(kv._internal_kv_list("gcs://")) == 0
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_exclusion(ray_start_cluster_head, working_dir, client_mode):
     cluster = ray_start_cluster_head
@@ -373,7 +372,7 @@ def test_exclusion(ray_start_cluster_head, working_dir, client_mode):
         "Test,FAILED,Test,FAILED,FAILED,Test,FAILED,FAILED"
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_exclusion_2(ray_start_cluster_head, working_dir, client_mode):
     cluster = ray_start_cluster_head
@@ -437,7 +436,7 @@ def create_file(p):
         "FAILED,Test,Test,FAILED,FAILED,Test,Test,FAILED,FAILED,FAILED,FAILED"
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_runtime_env_getter(ray_start_cluster_head, working_dir, client_mode):
     cluster = ray_start_cluster_head
@@ -452,7 +451,7 @@ def test_runtime_env_getter(ray_start_cluster_head, working_dir, client_mode):
     assert out.strip().split()[-1] == working_dir
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_two_node_uri(two_node_cluster, working_dir, client_mode):
     cluster, _ = two_node_cluster
@@ -476,7 +475,7 @@ def test_two_node_uri(two_node_cluster, working_dir, client_mode):
     assert len(kv._internal_kv_list("pingcs://")) == 1
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_regular_actors(ray_start_cluster_head, working_dir, client_mode):
     cluster = ray_start_cluster_head
@@ -494,7 +493,7 @@ def test_regular_actors(ray_start_cluster_head, working_dir, client_mode):
     assert len(kv._internal_kv_list("gcs://")) == 0
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 @pytest.mark.parametrize("client_mode", [True, False])
 def test_detached_actors(ray_start_cluster_head, working_dir, client_mode):
     cluster = ray_start_cluster_head
@@ -523,7 +522,7 @@ def test_detached_actors(ray_start_cluster_head, working_dir, client_mode):
     assert len(kv._internal_kv_list("gcs://")) == 0
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 def test_jobconfig_compatible_1(ray_start_cluster_head, working_dir):
     # start job_config=None
     # start job_config=something
@@ -549,7 +548,7 @@ def test_jobconfig_compatible_1(ray_start_cluster_head, working_dir):
     proc.wait()
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 def test_jobconfig_compatible_2(ray_start_cluster_head, working_dir):
     # start job_config=something
     # start job_config=None
@@ -574,7 +573,7 @@ def test_jobconfig_compatible_2(ray_start_cluster_head, working_dir):
     proc.wait()
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 def test_jobconfig_compatible_3(ray_start_cluster_head, working_dir):
     # start job_config=something
     # start job_config=something else
@@ -600,7 +599,7 @@ def test_jobconfig_compatible_3(ray_start_cluster_head, working_dir):
     assert out.strip().split()[-1] == "OK"
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 def test_util_without_job_config(shutdown_only):
     from ray.cluster_utils import Cluster
 
@@ -636,7 +635,7 @@ def run():
         os.chdir(old_dir)
 
 
-@unittest.skipIf(sys.platform == "win32", "Fail to create temp dir.")
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
 def test_init(shutdown_only):
     with tempfile.TemporaryDirectory() as tmp_dir:
         old_dir = os.getcwd()