mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-06-08 15:05:17 +02:00
refine tests
This commit is contained in:
parent
665ddba1c0
commit
78b7e164f9
2 changed files with 6 additions and 14 deletions
|
|
@ -23,7 +23,7 @@ import sys
|
|||
import traceback
|
||||
import typing
|
||||
from pathlib import Path
|
||||
from typing import Any, List, Tuple, Union, get_args, get_origin
|
||||
from typing import Any, List, Tuple, Union
|
||||
|
||||
import aiofiles
|
||||
import loguru
|
||||
|
|
@ -147,19 +147,7 @@ class OutputParser:
|
|||
if extracted_content:
|
||||
return extracted_content.group(1).strip()
|
||||
else:
|
||||
return "No content found between [CONTENT] and [/CONTENT] tags."
|
||||
|
||||
@staticmethod
|
||||
def is_supported_list_type(i):
|
||||
origin = get_origin(i)
|
||||
if origin is not List:
|
||||
return False
|
||||
|
||||
args = get_args(i)
|
||||
if args == (str,) or args == (Tuple[str, str],) or args == (List[str],):
|
||||
return True
|
||||
|
||||
return False
|
||||
raise ValueError(f"Could not find content between [{tag}] and [/{tag}]")
|
||||
|
||||
@classmethod
|
||||
def parse_data_with_mapping(cls, data, mapping):
|
||||
|
|
|
|||
|
|
@ -91,6 +91,10 @@ class TestGetProjectRoot:
|
|||
x=(TutorialAssistant, RunCode(), "a"),
|
||||
want={"metagpt.roles.tutorial_assistant.TutorialAssistant", "metagpt.actions.run_code.RunCode", "a"},
|
||||
),
|
||||
Input(
|
||||
x={"a": TutorialAssistant, "b": RunCode(), "c": "a"},
|
||||
want={"a", "metagpt.roles.tutorial_assistant.TutorialAssistant", "metagpt.actions.run_code.RunCode"},
|
||||
),
|
||||
]
|
||||
for i in inputs:
|
||||
v = any_to_str_set(i.x)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue