diff options
Diffstat (limited to 'tests/amt/test_models.py')
| -rw-r--r-- | tests/amt/test_models.py | 36 |
1 files changed, 21 insertions, 15 deletions
diff --git a/tests/amt/test_models.py b/tests/amt/test_models.py index c2a61b5..cecd948 100644 --- a/tests/amt/test_models.py +++ b/tests/amt/test_models.py @@ -22,20 +22,26 @@ def get_assignment_response_bad_tsid( ) return res -def test_get_assignment(get_assignment_response): - assignment = Assignment.from_amt_get_assignment( - get_assignment_response["Assignment"] - ) - assert assignment.tsid is not None -def test_get_assignment_no_tsid(get_assignment_response_no_tsid): - assignment = Assignment.from_amt_get_assignment( - get_assignment_response_no_tsid["Assignment"] - ) - assert assignment.tsid is None +class TestAssignment: -def test_get_assignment_bad_tsid(get_assignment_response_bad_tsid): - assignment = Assignment.from_amt_get_assignment( - get_assignment_response_bad_tsid["Assignment"] - ) - assert assignment.tsid is None
\ No newline at end of file + @pytest.mark.anyio + def test_get_assignment(get_assignment_response): + assignment = Assignment.from_amt_get_assignment( + get_assignment_response["Assignment"] + ) + assert assignment.tsid is not None + + @pytest.mark.anyio + def test_get_assignment_no_tsid(get_assignment_response_no_tsid): + assignment = Assignment.from_amt_get_assignment( + get_assignment_response_no_tsid["Assignment"] + ) + assert assignment.tsid is None + + @pytest.mark.anyio + def test_get_assignment_bad_tsid(get_assignment_response_bad_tsid): + assignment = Assignment.from_amt_get_assignment( + get_assignment_response_bad_tsid["Assignment"] + ) + assert assignment.tsid is None |