Skip to content

Commit

Permalink
[xray] Re-enable some stress tests and convert stress_tests to pytest. (
Browse files Browse the repository at this point in the history
ray-project#2285)

* Fix one of the stress tests, fix ray.global_state.client_table when called early on.

* Re-enable testWait.

* Convert stress_tests.py to pytest.

* Fix
  • Loading branch information
robertnishihara authored and pcmoritz committed Jul 7, 2018
1 parent 3a97289 commit e3534c4
Show file tree
Hide file tree
Showing 3 changed files with 516 additions and 514 deletions.
4 changes: 2 additions & 2 deletions .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ matrix:
- python test/tensorflow_test.py
- python test/failure_test.py
- python test/microbenchmarks.py
- python test/stress_tests.py
- python -m pytest test/stress_tests.py
# - python test/component_failures_test.py
- python test/multi_node_test.py
- python test/recursion_test.py
Expand Down Expand Up @@ -194,7 +194,7 @@ script:
- python test/tensorflow_test.py
- python test/failure_test.py
- python test/microbenchmarks.py
- python test/stress_tests.py
- python -m pytest test/stress_tests.py
- python test/component_failures_test.py
- python test/multi_node_test.py
- python test/recursion_test.py
Expand Down
6 changes: 6 additions & 0 deletions python/ray/experimental/state.py
Original file line number Diff line number Diff line change
Expand Up @@ -500,6 +500,12 @@ def client_table(self):
message = self.redis_client.execute_command(
"RAY.TABLE_LOOKUP", ray.gcs_utils.TablePrefix.CLIENT, "",
NIL_CLIENT_ID)

# Handle the case where no clients are returned. This should only
# occur potentially immediately after the cluster is started.
if message is None:
return []

node_info = []
gcs_entry = ray.gcs_utils.GcsTableEntry.GetRootAsGcsTableEntry(
message, 0)
Expand Down
Loading

0 comments on commit e3534c4

Please sign in to comment.