Skip to content

Commit

Permalink
Update providers/standard/tests/provider_tests/standard/utils/test_py…
Browse files Browse the repository at this point in the history
…thon_virtualenv.py

Co-authored-by: Wei Lee <[email protected]>
  • Loading branch information
2 people authored and potiuk committed Feb 6, 2025
1 parent 4c515fa commit 696e0ea
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -213,8 +213,8 @@ def f():
import funcsigs
"""
)
py_source = decorator + SCRIPT
expected_source = expected_decorator + SCRIPT if expected_decorator else SCRIPT.lstrip()
py_source = f"{decorator}{SCRIPT}"
expected_source = f"{expected_decorator}{SCRIPT}" if expected_decorator else SCRIPT.lstrip()

res = remove_task_decorator(python_source=py_source, task_decorator_name="@task.virtualenv")
assert res == expected_source

0 comments on commit 696e0ea

Please sign in to comment.