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

allow driver to run in PYTHON_MODE, which is equivalent to serial Python

Browse Source
This commit is contained in:
Robert Nishihara
2016-06-26 13:43:54 -07:00
parent e8e378883a
commit d97bce0d64
5 changed files with 44 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/runtest.py
+21
View File
@@ -377,5 +377,26 @@ class ReferenceCountingTest(unittest.TestCase):
services.cleanup()
class PythonModeTest(unittest.TestCase):
def testObjRefAliasing(self):
services.start_singlenode_cluster(driver_mode=ray.PYTHON_MODE)
xref = test_functions.test_alias_h()
self.assertTrue(np.alltrue(xref == np.ones([3, 4, 5]))) # remote functions should return by value
self.assertTrue(np.alltrue(xref == ray.get(xref))) # ray.get should be the identity
y = np.random.normal(size=[11, 12])
self.assertTrue(np.alltrue(y == ray.put(y))) # ray.put should be the identity
# make sure objects are immutable, this example is why we need to copy
# arguments before passing them into remote functions in python mode
aref = test_functions.python_mode_f()
self.assertTrue(np.alltrue(aref == np.array([0, 0])))
bref = test_functions.python_mode_g(aref)
self.assertTrue(np.alltrue(aref == np.array([0, 0]))) # python_mode_g should not mutate aref
self.assertTrue(np.alltrue(bref == np.array([1, 0])))
services.cleanup()
if __name__ == "__main__":
unittest.main()