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

implement keyword arguments

Browse Source
This commit is contained in:
Philipp Moritz
2016-06-03 00:10:17 -07:00
parent 5ca5f7502b
commit 429d9025eb
4 changed files with 59 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/runtest.py
+37
View File
@@ -181,6 +181,43 @@ class APITest(unittest.TestCase):
services.cleanup()
def testKeywordArgs(self):
test_dir = os.path.dirname(os.path.abspath(__file__))
test_path = os.path.join(test_dir, "testrecv.py")
services.start_singlenode_cluster(return_drivers=False, num_workers_per_objstore=3, worker_path=test_path)
x = test_functions.keyword_fct1(1)
self.assertEqual(orchpy.pull(x), "1 hello")
x = test_functions.keyword_fct1(1, "hi")
self.assertEqual(orchpy.pull(x), "1 hi")
x = test_functions.keyword_fct1(1, b="world")
self.assertEqual(orchpy.pull(x), "1 world")
x = test_functions.keyword_fct2(a="w", b="hi")
self.assertEqual(orchpy.pull(x), "w hi")
x = test_functions.keyword_fct2(b="hi", a="w")
self.assertEqual(orchpy.pull(x), "w hi")
x = test_functions.keyword_fct2(a="w")
self.assertEqual(orchpy.pull(x), "w world")
x = test_functions.keyword_fct2(b="hi")
self.assertEqual(orchpy.pull(x), "hello hi")
x = test_functions.keyword_fct2("w")
self.assertEqual(orchpy.pull(x), "w world")
x = test_functions.keyword_fct2("w", "hi")
self.assertEqual(orchpy.pull(x), "w hi")
x = test_functions.keyword_fct3(0, 1, c="w", d="hi")
self.assertEqual(orchpy.pull(x), "0 1 w hi")
x = test_functions.keyword_fct3(0, 1, d="hi", c="w")
self.assertEqual(orchpy.pull(x), "0 1 w hi")
x = test_functions.keyword_fct3(0, 1, c="w")
self.assertEqual(orchpy.pull(x), "0 1 w world")
x = test_functions.keyword_fct3(0, 1, d="hi")
self.assertEqual(orchpy.pull(x), "0 1 hello hi")
x = test_functions.keyword_fct3(0, 1)
self.assertEqual(orchpy.pull(x), "0 1 hello world")
services.cleanup()
class ReferenceCountingTest(unittest.TestCase):
def testDeallocation(self):
test/test_functions.py
+14
View File
@@ -38,3 +38,17 @@ def empty_function():
@orchpy.distributed([], [int])
def trivial_function():
return 1
# Test keyword arguments
@orchpy.distributed([int, str], [str])
def keyword_fct1(a, b="hello"):
return "{} {}".format(a, b)
@orchpy.distributed([str, str], [str])
def keyword_fct2(a="hello", b="world"):
return "{} {}".format(a, b)
@orchpy.distributed([int, int, str, str], [str])
def keyword_fct3(a, b, c="hello", d="world"):
return "{} {} {} {}".format(a, b, c, d)