aboutsummaryrefslogtreecommitdiff
path: root/tests/incite/mergers/foundations/test_enriched_session.py
diff options
context:
space:
mode:
authorMax Nanis2026-09-01 10:43:29 -0700
committerMax Nanis2026-09-01 10:43:29 -0700
commit89ed44f466dc9a93d6f85931fb6eea0e9cbd27f6 (patch)
tree8bcce59aa74271e38f056c954ee3e8457dae241a /tests/incite/mergers/foundations/test_enriched_session.py
parentd19438ec4ccbbe4415c286c9ae89e3e5706ac553 (diff)
downloadgeneralresearch-89ed44f466dc9a93d6f85931fb6eea0e9cbd27f6.tar.gz
generalresearch-89ed44f466dc9a93d6f85931fb6eea0e9cbd27f6.zip
working through circular imports, removing __init__ loaded Types / Definitions
Diffstat (limited to 'tests/incite/mergers/foundations/test_enriched_session.py')
-rw-r--r--tests/incite/mergers/foundations/test_enriched_session.py31
1 files changed, 17 insertions, 14 deletions
diff --git a/tests/incite/mergers/foundations/test_enriched_session.py b/tests/incite/mergers/foundations/test_enriched_session.py
index 2a161e4..71b2442 100644
--- a/tests/incite/mergers/foundations/test_enriched_session.py
+++ b/tests/incite/mergers/foundations/test_enriched_session.py
@@ -4,29 +4,32 @@ from collections.abc import Callable
from datetime import UTC, datetime, timedelta
from decimal import Decimal
from itertools import product
+from typing import TYPE_CHECKING
import dask.dataframe as dd
import pandas as pd
import pytest
from dask.distributed import Client as DaskClient
-from generalresearch.incite.collections.thl_web import (
- SessionDFCollection,
- WallDFCollection,
-)
-from generalresearch.incite.mergers.foundations.enriched_session import (
- EnrichedSessionMerge,
-)
from generalresearch.incite.schemas.admin_responses import (
AdminPOPSessionSchema,
)
-from generalresearch.models.admin.request import (
- ReportRequest,
-)
-from generalresearch.models.thl.product import Product
-from generalresearch.models.thl.session import Session
-from generalresearch.models.thl.user import User
-from generalresearch.pg_helper import PostgresConfig
+
+if TYPE_CHECKING:
+ from generalresearch.incite.collections.thl_web import (
+ SessionDFCollection,
+ WallDFCollection,
+ )
+ from generalresearch.incite.mergers.foundations.enriched_session import (
+ EnrichedSessionMerge,
+ )
+ from generalresearch.models.admin.request import (
+ ReportRequest,
+ )
+ from generalresearch.models.thl.product import Product
+ from generalresearch.models.thl.session import Session
+ from generalresearch.models.thl.user import User
+ from generalresearch.pg_helper import PostgresConfig
@pytest.mark.parametrize(