aboutsummaryrefslogtreecommitdiff
path: root/test_utils/incite/collections/conftest.py
diff options
context:
space:
mode:
authorMax Nanis2026-08-21 11:31:13 -0700
committerMax Nanis2026-08-21 11:31:13 -0700
commitff538deb290c851364df85aa6ccc64bc3719ef64 (patch)
treec04c3573ab6adab70971fef00d3420a5da5f9619 /test_utils/incite/collections/conftest.py
parent4fe0f6b5e0f0c744902e4c3ab8940e23a6f8a2e1 (diff)
parentc4a44873540ca4c0a3ab19b9beef4cfc6e0252a7 (diff)
downloadgeneralresearch-ff538deb290c851364df85aa6ccc64bc3719ef64.tar.gz
generalresearch-ff538deb290c851364df85aa6ccc64bc3719ef64.zip
Merge branch 'master' into dev
Diffstat (limited to 'test_utils/incite/collections/conftest.py')
-rw-r--r--test_utils/incite/collections/conftest.py38
1 files changed, 20 insertions, 18 deletions
diff --git a/test_utils/incite/collections/conftest.py b/test_utils/incite/collections/conftest.py
index 74e4081..88eef72 100644
--- a/test_utils/incite/collections/conftest.py
+++ b/test_utils/incite/collections/conftest.py
@@ -1,5 +1,7 @@
+from __future__ import annotations
+
from datetime import datetime, timedelta
-from typing import TYPE_CHECKING, Callable, Optional
+from typing import TYPE_CHECKING, Callable
import pytest
@@ -21,12 +23,12 @@ if TYPE_CHECKING:
@pytest.fixture
def user_collection(
- mnt_filepath: "GRLDatasets",
+ mnt_filepath: GRLDatasets,
offset: str,
duration: timedelta,
start: datetime,
thl_web_rr: PostgresConfig,
-) -> "UserDFCollection":
+) -> UserDFCollection:
from generalresearch.incite.collections.thl_web import (
DFCollectionType,
UserDFCollection,
@@ -43,12 +45,12 @@ def user_collection(
@pytest.fixture
def wall_collection(
- mnt_filepath: "GRLDatasets",
+ mnt_filepath: GRLDatasets,
offset: str,
duration: timedelta,
start: datetime,
thl_web_rr: PostgresConfig,
-) -> "WallDFCollection":
+) -> WallDFCollection:
from generalresearch.incite.collections.thl_web import (
DFCollectionType,
WallDFCollection,
@@ -65,12 +67,12 @@ def wall_collection(
@pytest.fixture
def session_collection(
- mnt_filepath: "GRLDatasets",
+ mnt_filepath: GRLDatasets,
offset: str,
duration: timedelta,
start: datetime,
thl_web_rr: PostgresConfig,
-) -> "SessionDFCollection":
+) -> SessionDFCollection:
from generalresearch.incite.collections.thl_web import (
DFCollectionType,
SessionDFCollection,
@@ -103,12 +105,12 @@ def session_collection(
@pytest.fixture
def task_adj_collection(
- mnt_filepath: "GRLDatasets",
+ mnt_filepath: GRLDatasets,
offset: str,
- duration: Optional[timedelta],
+ duration: timedelta | None,
start: datetime,
thl_web_rr: PostgresConfig,
-) -> "TaskAdjustmentDFCollection":
+) -> TaskAdjustmentDFCollection:
from generalresearch.incite.collections.thl_web import (
DFCollectionType,
TaskAdjustmentDFCollection,
@@ -127,12 +129,12 @@ def task_adj_collection(
@pytest.fixture
def auditlog_collection(
- mnt_filepath: "GRLDatasets",
+ mnt_filepath: GRLDatasets,
offset: str,
duration: timedelta,
start: datetime,
thl_web_rr: PostgresConfig,
-) -> "AuditLogDFCollection":
+) -> AuditLogDFCollection:
from generalresearch.incite.collections.thl_web import (
AuditLogDFCollection,
DFCollectionType,
@@ -149,12 +151,12 @@ def auditlog_collection(
@pytest.fixture
def ledger_collection(
- mnt_filepath: "GRLDatasets",
+ mnt_filepath: GRLDatasets,
offset: str,
duration: timedelta,
start: datetime,
thl_web_rr: PostgresConfig,
-) -> "LedgerDFCollection":
+) -> LedgerDFCollection:
from generalresearch.incite.collections.thl_web import (
DFCollectionType,
LedgerDFCollection,
@@ -171,7 +173,7 @@ def ledger_collection(
@pytest.fixture
def rm_ledger_collection(
- ledger_collection: "LedgerDFCollection",
+ ledger_collection: LedgerDFCollection,
) -> Callable[..., None]:
def _inner():
@@ -187,13 +189,13 @@ def rm_ledger_collection(
@pytest.fixture
def df_collection(
- mnt_filepath: "GRLDatasets",
- df_collection_data_type: "DFCollectionType",
+ mnt_filepath: GRLDatasets,
+ df_collection_data_type: DFCollectionType,
offset: str,
duration: timedelta,
utc_90days_ago: datetime,
thl_web_rr: PostgresConfig,
-) -> "DFCollection":
+) -> DFCollection:
from generalresearch.incite.collections import DFCollection
start = utc_90days_ago.replace(microsecond=0)