aboutsummaryrefslogtreecommitdiff
path: root/test_utils/conftest.py
diff options
context:
space:
mode:
authorMax Nanis2026-03-08 21:57:53 -0400
committerMax Nanis2026-03-08 21:57:53 -0400
commitce291a165fab6b6dc9f053c7b75a699d0fdf389f (patch)
tree7a7b393f432684e13ffe92c40015fb6afc07ceac /test_utils/conftest.py
parentb2e56ec3ebc5eb91eb10cc37cc9e02102f441660 (diff)
downloadgeneralresearch-ce291a165fab6b6dc9f053c7b75a699d0fdf389f.tar.gz
generalresearch-ce291a165fab6b6dc9f053c7b75a699d0fdf389f.zip
Simple typing changes, Ruff import formatter. p2
Diffstat (limited to 'test_utils/conftest.py')
-rw-r--r--test_utils/conftest.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/test_utils/conftest.py b/test_utils/conftest.py
index e074301..54fb682 100644
--- a/test_utils/conftest.py
+++ b/test_utils/conftest.py
@@ -247,13 +247,16 @@ def utc_30days_ago() -> "datetime":
@pytest.fixture(scope="function")
-def delete_df_collection(thl_web_rw, create_main_accounts) -> Callable:
+def delete_df_collection(
+ thl_web_rw: PostgresConfig, create_main_accounts: Callable[..., None]
+) -> Callable[..., None]:
+
from generalresearch.incite.collections import (
DFCollection,
DFCollectionType,
)
- def _teardown_events(coll: "DFCollection"):
+ def _inner(coll: "DFCollection"):
match coll.data_type:
case DFCollectionType.LEDGER:
for table in [
@@ -290,7 +293,7 @@ def delete_df_collection(thl_web_rw, create_main_accounts) -> Callable:
query=f"DELETE FROM {coll.data_type.value};",
)
- return _teardown_events
+ return _inner
# === GR Related ===