mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 06:47:13 +08:00
Simon Mo
31 lines
769 B
Python
31 lines
769 B
Python
import numpy as np
|
|
import unittest
|
|
|
|
import ray
|
|
from ray.test_utils import put_unpinned_object
|
|
|
|
|
|
class TestObjectLostErrors(unittest.TestCase):
|
|
def setUp(self):
|
|
ray.init(
|
|
num_cpus=1,
|
|
object_store_memory=150 * 1024 * 1024,
|
|
_redis_max_memory=10**8)
|
|
|
|
def tearDown(self):
|
|
ray.shutdown()
|
|
|
|
def testDriverPutEvictedCannotReconstruct(self):
|
|
x_id = put_unpinned_object(np.zeros(1 * 1024 * 1024))
|
|
ray.get(x_id)
|
|
for _ in range(20):
|
|
ray.put(np.zeros(10 * 1024 * 1024))
|
|
self.assertRaises(ray.exceptions.ObjectLostError,
|
|
lambda: ray.get(x_id))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import pytest
|
|
import sys
|
|
sys.exit(pytest.main(["-v", __file__]))
|