aboutsummaryrefslogtreecommitdiff
path: root/tests/amt/test_models.py
diff options
context:
space:
mode:
authorMax Nanis2026-02-26 15:51:49 -0500
committerMax Nanis2026-02-26 15:51:49 -0500
commit0bf32fadd85d5938ae29d489efdd82e2cd137300 (patch)
tree814e8128947fb604dc7cc3509e72260d95757590 /tests/amt/test_models.py
parent04aee0dc7e908ce020d2d2c3f8ffb4a96424b883 (diff)
downloadamt-jb-0bf32fadd85d5938ae29d489efdd82e2cd137300.tar.gz
amt-jb-0bf32fadd85d5938ae29d489efdd82e2cd137300.zip
Passing Managers into flow tasks for better pytest usage. Conftests broken out into seperate fixture files. Extensive type hinting.
Diffstat (limited to 'tests/amt/test_models.py')
-rw-r--r--tests/amt/test_models.py47
1 files changed, 0 insertions, 47 deletions
diff --git a/tests/amt/test_models.py b/tests/amt/test_models.py
deleted file mode 100644
index cecd948..0000000
--- a/tests/amt/test_models.py
+++ /dev/null
@@ -1,47 +0,0 @@
-import copy
-
-import pytest
-
-from jb.models.assignment import Assignment
-
-
-@pytest.fixture
-def get_assignment_response_bad_tsid(
- get_assignment_response, amt_worker_id, amt_assignment_id
-):
- res = copy.deepcopy(get_assignment_response)
- res["Assignment"]["Answer"] = (
- '<?xml version="1.0" encoding="UTF-8"?>\n'
- '<QuestionFormAnswers xmlns="http://mechanicalturk.amazonaws.com/AWSMechanicalTurkDataSchemas/2005-10-01/QuestionFormAnswers.xsd">\n '
- "<Answer>\n <QuestionIdentifier>amt_worker_id</QuestionIdentifier>\n "
- f" <FreeText>{amt_worker_id}</FreeText>\n </Answer>\n <Answer>\n "
- " <QuestionIdentifier>amt_assignment_id</QuestionIdentifier>\n "
- f" <FreeText>{amt_assignment_id}</FreeText>\n </Answer>\n "
- f" <Answer>\n <QuestionIdentifier>tsid</QuestionIdentifier>\n abc123 <FreeText></FreeText>\n </Answer>\n"
- f"</QuestionFormAnswers>"
- )
- return res
-
-
-class TestAssignment:
-
- @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