aboutsummaryrefslogtreecommitdiff
path: root/test_utils/models
diff options
context:
space:
mode:
authorstuppie2026-03-09 18:44:43 -0600
committerstuppie2026-03-09 18:44:43 -0600
commit459a8638924a6bc4c98eddee317e2dca59ed054b (patch)
tree683d434f2471b4dbcf0ae62d5550ee163940ec72 /test_utils/models
parent2f92429a68ec7209059d2d18fe67964c8dd57cf2 (diff)
downloadgeneralresearch-459a8638924a6bc4c98eddee317e2dca59ed054b.tar.gz
generalresearch-459a8638924a6bc4c98eddee317e2dca59ed054b.zip
Fix type hints
Diffstat (limited to 'test_utils/models')
-rw-r--r--test_utils/models/conftest.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/test_utils/models/conftest.py b/test_utils/models/conftest.py
index 468bea2..64bdec6 100644
--- a/test_utils/models/conftest.py
+++ b/test_utils/models/conftest.py
@@ -590,7 +590,7 @@ def ip_record_factory(
@pytest.fixture(scope="session")
-def buyer(buyer_manager: "BuyerManager") -> Buyer:
+def buyer(buyer_manager: "BuyerManager") -> "Buyer":
buyer_code = uuid4().hex
buyer_manager.bulk_get_or_create(source=Source.TESTING, codes=[buyer_code])
b = Buyer(
@@ -601,7 +601,7 @@ def buyer(buyer_manager: "BuyerManager") -> Buyer:
@pytest.fixture(scope="session")
-def buyer_factory(buyer_manager: "BuyerManager") -> Callable[..., Buyer]:
+def buyer_factory(buyer_manager: "BuyerManager") -> Callable[..., "Buyer"]:
def _inner() -> Buyer:
return buyer_manager.bulk_get_or_create(
@@ -612,7 +612,7 @@ def buyer_factory(buyer_manager: "BuyerManager") -> Callable[..., Buyer]:
@pytest.fixture(scope="session")
-def survey(survey_manager: "SurveyManager", buyer: "Buyer") -> Survey:
+def survey(survey_manager: "SurveyManager", buyer: "Buyer") -> "Survey":
s = Survey(source=Source.TESTING, survey_id=uuid4().hex, buyer_code=buyer.code)
survey_manager.create_bulk([s])
return s