From 5ffc84b803e2ffdaac9039c81c7a069df8f7c2f7 Mon Sep 17 00:00:00 2001 From: Janine vN Date: Fri, 11 Aug 2023 19:15:50 -0400 Subject: Add tearDown feature to unittests --- backend/routes/forms/unittesting.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'backend') 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):" -- cgit v1.2.3