diff --git a/tests/test_tools_patch.py b/tests/test_tools_patch.py index 24d183a2..11e44704 100644 --- a/tests/test_tools_patch.py +++ b/tests/test_tools_patch.py @@ -44,7 +44,19 @@ def hello(name="world"): ) -def test_clear_file(): +def test_apply_clear_file(): + content = "test" + codeblock = """ +<<<<<<< ORIGINAL +test +======= +>>>>>>> UPDATED + """ + result = apply(codeblock, content) + assert result == "" + + +def test_apply_rm_function(): # only remove code in patch content = """ def hello():