def test_actor_method_fail(ray_start_2_cpus):
ag = ActorGroup(actor_cls=DummyActor, num_actors=2)
with pytest.raises(TypeError):
ag.return_arg(1)
with pytest.raises(AttributeError):
ag.non_existent_method.remote()
def test_actor_creation_num_cpus(ray_start_2_cpus):
assert ray.available_resources()["CPU"] == 2
ag = ActorGroup(actor_cls=DummyActor, num_cpus_per_actor=2)
time.sleep(1)
assert len(ag) == 1
# Make sure both CPUs are being used by the actor.
assert "CPU" not in ray.available_resources()
ag.shutdown()
def test_bad_resources(ray_start_2_cpus):
with pytest.raises(ValueError):
ActorGroup(actor_cls=DummyActor, num_actors=-1)
with pytest.raises(ValueError):
ActorGroup(actor_cls=DummyActor, num_actors=-1)
with pytest.raises(ValueError):
ActorGroup(actor_cls=DummyActor, num_actors=-1)
def test_actor_shutdown(ray_start_2_cpus):
assert ray.available_resources()["CPU"] == 2
ag = ActorGroup(actor_cls=DummyActor, num_actors=2)
time.sleep(1)
assert "CPU" not in ray.available_resources()
assert len(ray.state.actors()) == 2
ag.shutdown()
time.sleep(1)
assert ray.available_resources()["CPU"] == 2
with pytest.raises(RuntimeError):
ag.return_arg.remote(1)
def test_actor_restart(ray_start_2_cpus):
ag = ActorGroup(actor_cls=DummyActor, num_actors=2)
with pytest.raises(RuntimeError):
ag.start()
# Avoid race condition.
time.sleep(1)
ag.shutdown(0)
ag.start()
ray.get(ag.return_arg.remote(1))
def test_actor_metadata(ray_start_2_cpus):
ag = ActorGroup(actor_cls=DummyActor, num_actors=2)
assert ag.actor_metadata == ["metadata", "metadata"]
def test_actor_method(ray_start_2_cpus):
ag = ActorGroup(actor_cls=DummyActor, num_actors=2)
assert ray.get(ag.return_arg.remote(1)) == [1, 1]