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

[Autoscaler] Pass custom resources to "ray start" multi instance autoscaling (#9986)

Browse Source
This commit is contained in:
Alex Wu
2020-08-17 22:34:07 -07:00
committed by GitHub
parent 71f6f83f1d
commit 0b5d5ec17d
11 changed files with 303 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_autoscaler.py
+23 -10
View File
@@ -51,18 +51,31 @@ class MockProcessRunner:
self.check_call(cmd)
return "command-output".encode()
def assert_has_call(self, ip, pattern):
def assert_has_call(self, ip, pattern=None, exact=None):
assert pattern or exact, \
"Must specify either a pattern or exact match."
out = ""
for cmd in self.calls:
msg = " ".join(cmd)
if ip in msg:
out += msg
out += "\n"
if pattern in out:
return True
if pattern is not None:
for cmd in self.calls:
msg = " ".join(cmd)
if ip in msg:
out += msg
out += "\n"
if pattern in out:
return True
else:
raise Exception("Did not find [{}] in [{}] for {}".format(
pattern, out, ip))
else:
raise Exception("Did not find [{}] in [{}] for {}".format(
pattern, out, ip))
for cmd in self.calls:
msg = " ".join(cmd)
if ip in msg:
out += msg
out += "\n"
if cmd == exact:
return True
raise Exception("Did not find {} in {} for {}".format(
exact, out, ip))
def assert_not_has_call(self, ip, pattern):
out = ""