diff --git a/tests/unit/test_va.py b/tests/unit/test_va.py
index 8f5e234d..3fe619e8 100644
--- a/tests/unit/test_va.py
+++ b/tests/unit/test_va.py
@@ -30,9 +30,7 @@ def test_parse_execution_no_test_multi_plan_edit():
def test_parse_execution_custom_tool_names_generate():
code = "generate_vision_code(artifacts, 'code.py', 'Generate code', ['image.png'])"
assert (
- parse_execution(
- code, test_multi_plan=False, custom_tool_names=["owl_v2_image"]
- )
+ parse_execution(code, test_multi_plan=False, custom_tool_names=["owl_v2_image"])
== "generate_vision_code(artifacts, 'code.py', 'Generate code', ['image.png'], test_multi_plan=False, custom_tool_names=['owl_v2_image'])"
)
@@ -40,9 +38,7 @@ def test_parse_execution_custom_tool_names_generate():
def test_parse_execution_custom_tool_names_edit():
code = "edit_vision_code(artifacts, 'code.py', ['Generate code'], ['image.png'])"
assert (
- parse_execution(
- code, test_multi_plan=False, custom_tool_names=["owl_v2_image"]
- )
+ parse_execution(code, test_multi_plan=False, custom_tool_names=["owl_v2_image"])
== "edit_vision_code(artifacts, 'code.py', ['Generate code'], ['image.png'], test_multi_plan=False, custom_tool_names=['owl_v2_image'])"
)