wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-03 09:45:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor pytest fixtures for ray core (#4390)

Browse Source
This commit is contained in:
Yuhong Guo
2019-03-20 11:48:32 +08:00
committed by Hao Chen
parent c6f15a0057
commit 8ce7565530
22 changed files with 378 additions and 681 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_node_manager.py
+2 -15
View File
@@ -2,27 +2,14 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import pytest
import ray
from ray.tests.cluster_utils import Cluster
from ray.tests.utils import run_string_as_driver
@pytest.fixture()
def ray_start_empty_cluster():
cluster = Cluster()
yield cluster
# The code after the yield will run as teardown code.
ray.shutdown()
cluster.shutdown()
# This tests the queue transitions for infeasible tasks. This has been an issue
# in the past, e.g., https://github.com/ray-project/ray/issues/3275.
def test_infeasible_tasks(ray_start_empty_cluster):
cluster = ray_start_empty_cluster
def test_infeasible_tasks(ray_start_cluster):
cluster = ray_start_cluster
@ray.remote
def f():