From 8c1940445503fd6678d0961600f2be81622793a2 Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Tue, 24 Feb 2026 17:26:15 -0500
Subject: Extensive use of type checking. Movement of pytest conf towards
 handling managers (for db agnostic unittest). Starting to organize pytests.

---
 tests/amt/test_models.py | 36 +++++++++++++++++++++---------------
 1 file changed, 21 insertions(+), 15 deletions(-)

(limited to 'tests/amt/test_models.py')

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
-- 
cgit v1.2.3