mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 08:23:44 +08:00
Edward Oakes
30 lines
740 B
Python
30 lines
740 B
Python
import numpy as np
|
|
import unittest
|
|
|
|
import ray
|
|
|
|
|
|
class TestUnreconstructableErrors(unittest.TestCase):
|
|
def setUp(self):
|
|
ray.init(
|
|
num_cpus=1,
|
|
object_store_memory=150 * 1024 * 1024,
|
|
redis_max_memory=10000000)
|
|
|
|
def tearDown(self):
|
|
ray.shutdown()
|
|
|
|
def testDriverPutEvictedCannotReconstruct(self):
|
|
x_id = ray.put(np.zeros(1 * 1024 * 1024), weakref=True)
|
|
ray.get(x_id)
|
|
for _ in range(20):
|
|
ray.put(np.zeros(10 * 1024 * 1024))
|
|
self.assertRaises(ray.exceptions.UnreconstructableError,
|
|
lambda: ray.get(x_id))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import pytest
|
|
import sys
|
|
sys.exit(pytest.main(["-v", __file__]))
|