diff options
| -rw-r--r-- | backend/routes/forms/unittesting.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/backend/routes/forms/unittesting.py b/backend/routes/forms/unittesting.py index 1b042e4..a02afea 100644 --- a/backend/routes/forms/unittesting.py +++ b/backend/routes/forms/unittesting.py @@ -43,6 +43,8 @@ def _make_unit_code(units: dict[str, str]) -> str: # Function definition if unit_name == "setUp": result += "\ndef setUp(self):" + elif unit_name == "tearDown": + result += "\ndef tearDown(self):" else: name = f"test_{unit_name.removeprefix('#').removeprefix('test_')}" result += f"\nasync def {name}(self):" |