diff options
| author | Max Nanis | 2026-08-21 01:38:27 -0700 |
|---|---|---|
| committer | Max Nanis | 2026-08-21 01:38:27 -0700 |
| commit | 05102628a7dc85a5a19a32415a7ec41ea7a91812 (patch) | |
| tree | f5253ad2f57a9f05776eaf8f5dfb21bedbd08c7e /test_utils/incite/conftest.py | |
| parent | 0c10237223829818752ab4fc431957c0b22a0e23 (diff) | |
| download | generalresearch-05102628a7dc85a5a19a32415a7ec41ea7a91812.tar.gz generalresearch-05102628a7dc85a5a19a32415a7ec41ea7a91812.zip | |
moving create_dummy, organizing test_utils, working on django config for any repos
Diffstat (limited to 'test_utils/incite/conftest.py')
| -rw-r--r-- | test_utils/incite/conftest.py | 34 |
1 files changed, 17 insertions, 17 deletions
diff --git a/test_utils/incite/conftest.py b/test_utils/incite/conftest.py index 0e2f7bd..12e57c5 100644 --- a/test_utils/incite/conftest.py +++ b/test_utils/incite/conftest.py @@ -1,9 +1,11 @@ +from __future__ import annotations + from datetime import datetime, timedelta, timezone from os.path import join as pjoin from pathlib import Path from random import choice as randchoice from shutil import rmtree -from typing import TYPE_CHECKING, Callable, Optional +from typing import TYPE_CHECKING, Callable from uuid import uuid4 import pytest @@ -30,7 +32,7 @@ fake = Faker() @pytest.fixture -def mnt_gr_api_dir(request: SubRequest, settings: "GRLBaseSettings") -> Path: +def mnt_gr_api_dir(request: SubRequest, settings: GRLBaseSettings) -> Path: p = Path(settings.mnt_gr_api_dir) p.mkdir(parents=True, exist_ok=True) @@ -53,7 +55,7 @@ def mnt_gr_api_dir(request: SubRequest, settings: "GRLBaseSettings") -> Path: @pytest.fixture -def event_report_request(utc_hour_ago: datetime, start: datetime) -> "ReportRequest": +def event_report_request(utc_hour_ago: datetime, start: datetime) -> ReportRequest: from generalresearch.models.admin.request import ( ReportRequest, ReportType, @@ -69,7 +71,7 @@ def event_report_request(utc_hour_ago: datetime, start: datetime) -> "ReportRequ @pytest.fixture -def session_report_request(utc_hour_ago: datetime, start: datetime) -> "ReportRequest": +def session_report_request(utc_hour_ago: datetime, start: datetime) -> ReportRequest: from generalresearch.models.admin.request import ( ReportRequest, ReportType, @@ -85,7 +87,7 @@ def session_report_request(utc_hour_ago: datetime, start: datetime) -> "ReportRe @pytest.fixture -def mnt_filepath(request: SubRequest) -> "GRLDatasets": +def mnt_filepath(request: SubRequest) -> GRLDatasets: """ Creates a temporary file path for all DFCollections & Mergers parquet files. @@ -111,7 +113,7 @@ def mnt_filepath(request: SubRequest) -> "GRLDatasets": @pytest.fixture -def start(utc_90days_ago: datetime) -> "datetime": +def start(utc_90days_ago: datetime) -> datetime: s = utc_90days_ago.replace(microsecond=0) return s @@ -122,19 +124,19 @@ def offset() -> str: @pytest.fixture -def duration() -> Optional["timedelta"]: +def duration() -> timedelta | None: return timedelta(hours=1) @pytest.fixture -def df_collection_data_type() -> "DFCollectionType": +def df_collection_data_type() -> DFCollectionType: from generalresearch.incite.collections import DFCollectionType return DFCollectionType.TEST @pytest.fixture -def merge_type() -> "MergeType": +def merge_type() -> MergeType: from generalresearch.incite.mergers import MergeType return MergeType.TEST @@ -142,16 +144,16 @@ def merge_type() -> "MergeType": @pytest.fixture def incite_item_factory( - session_factory: Callable[..., "Session"], - product: "Product", - user_factory: Callable[..., "User"], - session_with_tx_factory: Callable[..., "Session"], + session_factory: Callable[..., Session], + product: Product, + user_factory: Callable[..., User], + session_with_tx_factory: Callable[..., Session], ) -> Callable[..., None]: def _inner( - item: "DFCollectionItem", + item: DFCollectionItem, observations: int = 3, - user: Optional["User"] = None, + user: User | None = None, ): from generalresearch.incite.collections import ( DFCollection, @@ -201,6 +203,4 @@ def incite_item_factory( case _: raise ValueError("Unsupported DFCollectionItem") - return None - return _inner |
