wehub-resource-sync
34 lines
830 B
Python
34 lines
830 B
Python
import sys
|
|
|
|
import pytest
|
|
|
|
import ray
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
sys.platform == "win32",
|
|
reason="Failing on Windows, affected by protocolbuffers/protobuf#10075.",
|
|
)
|
|
def test_protobuf_compatibility(shutdown_only):
|
|
protobuf_4_21_0 = {"pip": ["protobuf==4.21.0"]}
|
|
protobuf_3_20_3 = {"pip": ["protobuf==3.20.3"]}
|
|
|
|
ray.init()
|
|
|
|
@ray.remote
|
|
def load_ray():
|
|
import google.protobuf
|
|
|
|
# verfiy this no longer crashes. see
|
|
# https://github.com/ray-project/ray/issues/25282
|
|
import ray # noqa
|
|
|
|
return google.protobuf.__version__
|
|
|
|
assert "4.21.0" == ray.get(load_ray.options(runtime_env=protobuf_4_21_0).remote())
|
|
assert "3.20.3" == ray.get(load_ray.options(runtime_env=protobuf_3_20_3).remote())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-sv", __file__]))
|