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

refine tests

Browse source
This commit is contained in:
geekan 2024-01-02 21:51:24 +08:00
parent 339d9de5c7
commit 665ddba1c0
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/metagpt/utils/test_output_parser.py
View file

@ -130,7 +130,7 @@ def test_parse_with_markdown_mapping():
"Requirement Pool": (List[Tuple[str, str]], ...),
"Anything UNCLEAR": (str, ...),
}
t_text1 = """[CONTENT]## Original Requirements:
t_text_with_content_tag = """[CONTENT]## Original Requirements:
The user wants to create a web-based version of the game "Fly Bird".
@ -199,7 +199,10 @@ The product should be a web-based version of the game "Fly Bird" that is engagin
There are no unclear points.
[/CONTENT]"""
d = OutputParser.parse_data_with_mapping(t_text1, OUTPUT_MAPPING)
t_text_raw = t_text_with_content_tag.replace("[CONTENT]", "").replace("[/CONTENT]", "")
d = OutputParser.parse_data_with_mapping(t_text_with_content_tag, OUTPUT_MAPPING)
import json
print(json.dumps(d))
assert d["Original Requirements"] == t_text_raw.split("## Original Requirements:")[1].split("##")[0].strip()