wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-05 21:42:09 +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_recursion.py
+1 -15
View File
@@ -6,23 +6,9 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import pytest
import ray
@pytest.fixture
def ray_start():
# Start ray instance
ray.init(num_cpus=1)
# Run test using this fixture
yield None
# Shutdown ray instance
ray.shutdown()
@ray.remote
def factorial(n):
if n == 0:
@@ -30,7 +16,7 @@ def factorial(n):
return n * ray.get(factorial.remote(n - 1))
def test_recursion(ray_start):
def test_recursion(ray_start_regular):
assert ray.get(factorial.remote(0)) == 1
assert ray.get(factorial.remote(1)) == 1
assert ray.get(factorial.remote(2)) == 2