From ce291a165fab6b6dc9f053c7b75a699d0fdf389f Mon Sep 17 00:00:00 2001 From: Max Nanis Date: Sun, 8 Mar 2026 21:57:53 -0400 Subject: Simple typing changes, Ruff import formatter. p2 --- test_utils/conftest.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'test_utils/conftest.py') 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 === -- cgit v1.2.3