diff options
| author | Max Nanis | 2026-03-08 21:57:53 -0400 |
|---|---|---|
| committer | Max Nanis | 2026-03-08 21:57:53 -0400 |
| commit | ce291a165fab6b6dc9f053c7b75a699d0fdf389f (patch) | |
| tree | 7a7b393f432684e13ffe92c40015fb6afc07ceac /test_utils/conftest.py | |
| parent | b2e56ec3ebc5eb91eb10cc37cc9e02102f441660 (diff) | |
| download | generalresearch-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.py | 9 |
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 === |
