Extensive use of type checking. Movement of pytest conf towards handling managers (for db agnostic unittest). Starting to organize pytests.

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