修改函数名称

This commit is contained in:
黄伟韬 2024-08-16 11:22:02 +08:00
parent 0a1c965933
commit b27ba8476f
2 changed files with 10 additions and 10 deletions

View file

@ -30,20 +30,20 @@ class TestRoleZeroContextBuilder:
assert result == [{"content": "Updated content"}]
def test_replace_example_content(self, context_builder, mocker):
mocker.patch.object(RoleZeroContextBuilder, "replace_content_of_example_tag", return_value="Replaced content")
mocker.patch.object(RoleZeroContextBuilder, "fill_experience", return_value="Replaced content")
result = context_builder.replace_example_content("Original text", "New example content")
assert result == "Replaced content"
context_builder.replace_content_of_example_tag.assert_called_once_with("Original text", "New example content")
context_builder.fill_experience.assert_called_once_with("Original text", "New example content")
def test_replace_content_of_example_tag(self):
def test_fill_experience(self):
text = "Start\n# Past Experience\n<experience>\n\n# Instruction\nEnd"
new_content = "New content"
result = RoleZeroContextBuilder.replace_content_of_example_tag(text, new_content)
result = RoleZeroContextBuilder.fill_experience(text, new_content)
expected = "Start\n# Past Experience\nNew content\n\n# Instruction\nEnd"
assert result == expected
def test_replace_content_of_example_tag_no_match(self):
def test_fill_experience_no_match(self):
text = "Start\nNo markers\nEnd"
new_content = "New content"
result = RoleZeroContextBuilder.replace_content_of_example_tag(text, new_content)
result = RoleZeroContextBuilder.fill_experience(text, new_content)
assert result == text