apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: unit tests

Browse source
This commit is contained in:
莘权 马 2023-11-10 17:14:58 +08:00
parent d9a7443e5a
commit 282a86bfa7
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/metagpt/test_role.py
View file

@ -60,7 +60,7 @@ async def test_react():
name=seed.name, profile=seed.profile, goal=seed.goal, constraints=seed.constraints, desc=seed.desc
)
role.subscribe({seed.subscription})
assert role._rc.watch == {seed.subscription}
assert role._rc.watch == set({})
assert role.name == seed.name
assert role.profile == seed.profile
assert role._setting.goal == seed.goal
@ -69,7 +69,7 @@ async def test_react():
assert role.is_idle
env = Environment()
env.add_role(role)
assert env.get_subscribed_tags(role) == {seed.subscription}
assert env.get_subscription(role) == {seed.subscription}
env.publish_message(Message(content="test", msg_to=seed.subscription))
assert not role.is_idle
while not env.is_idle:
@ -82,19 +82,19 @@ async def test_react():
assert role.is_idle
tag = uuid.uuid4().hex
role.subscribe({tag})
assert env.get_subscribed_tags(role) == {seed.subscription, tag}
assert env.get_subscription(role) == {tag}
@pytest.mark.asyncio
async def test_msg_to():
m = Message(content="a", msg_to=["a", MockRole, Message])
assert m.msg_to == {"a", get_class_name(MockRole), get_class_name(Message)}
m = Message(content="a", send_to=["a", MockRole, Message])
assert m.send_to == set({"a", get_class_name(MockRole), get_class_name(Message)})
m = Message(content="a", cause_by=MockAction, msg_to={"a", MockRole, Message})
assert m.msg_to == {"a", get_class_name(MockRole), get_class_name(Message), get_class_name(MockAction)}
m = Message(content="a", cause_by=MockAction, send_to={"a", MockRole, Message})
assert m.send_to == set({"a", get_class_name(MockRole), get_class_name(Message)})
m = Message(content="a", msg_to=("a", MockRole, Message))
assert m.msg_to == {"a", get_class_name(MockRole), get_class_name(Message)}
m = Message(content="a", send_to=("a", MockRole, Message))
assert m.send_to == set({"a", get_class_name(MockRole), get_class_name(Message)})
if __name__ == "__main__":