aboutsummaryrefslogtreecommitdiff
path: root/test_utils/incite/conftest.py
diff options
context:
space:
mode:
Diffstat (limited to 'test_utils/incite/conftest.py')
-rw-r--r--test_utils/incite/conftest.py34
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