aboutsummaryrefslogtreecommitdiffstats
path: root/backend/models/question.py
diff options
context:
space:
mode:
authorGravatar Hassan Abouelela <[email protected]>2022-06-24 06:01:41 +0400
committerGravatar GitHub <[email protected]>2022-06-24 06:01:41 +0400
commitc615219749e3deabe48943f763623f5d8545bf80 (patch)
treea4dd7185e44b96cd114a8f062c4c72c8d6b626c6 /backend/models/question.py
parentMerge pull request #173 from adriangb/asgi-middleware (diff)
parentMake Unittests Async (diff)
Merge pull request #179 from python-discord/add-setup
Allow Setup For Unittests
Diffstat (limited to 'backend/models/question.py')
-rw-r--r--backend/models/question.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/backend/models/question.py b/backend/models/question.py
index 5a1334a..201aa51 100644
--- a/backend/models/question.py
+++ b/backend/models/question.py
@@ -15,8 +15,10 @@ class Unittests(BaseModel):
@validator("tests")
def validate_tests(cls, value: _TESTS_TYPE) -> _TESTS_TYPE:
"""Confirm that at least one test exists in a test suite."""
- if isinstance(value, dict) and len(value.keys()) == 0:
- raise ValueError("Must have at least one test in a test suite.")
+ if isinstance(value, dict):
+ keys = len(value.keys()) - (1 if "setUp" in value.keys() else 0)
+ if keys == 0:
+ raise ValueError("Must have at least one test in a test suite.")
return value