merge main

This commit is contained in:
geekan 2024-01-11 19:18:08 +08:00
commit e6df55bdb9
8 changed files with 74 additions and 30 deletions

View file

@ -3,6 +3,7 @@
# @Desc : unittest of Role
import pytest
from metagpt.llm import HumanProvider
from metagpt.roles.role import Role
@ -12,5 +13,10 @@ def test_role_desc():
assert role.desc == "Best Seller"
def test_role_human():
role = Role(is_human=True)
assert isinstance(role.llm, HumanProvider)
if __name__ == "__main__":
pytest.main([__file__, "-s"])

View file

@ -5,25 +5,26 @@
@Author : mashenquan
@File : test_redis.py
"""
from unittest.mock import AsyncMock
import mock
import pytest
from pytest_mock import mocker
from metagpt.config2 import Config
from metagpt.utils.redis import Redis
async def async_mock_from_url(*args, **kwargs):
mock_client = mock.AsyncMock()
mock_client = AsyncMock()
mock_client.set.return_value = None
mock_client.get.side_effect = [b"test", b""]
return mock_client
@pytest.mark.asyncio
@mock.patch("aioredis.from_url", return_value=async_mock_from_url())
async def test_redis(i):
redis = Config.default().redis
mocker.patch("aioredis.from_url", return_value=async_mock_from_url())
conn = Redis(redis)
await conn.set("test", "test", timeout_sec=0)