aboutsummaryrefslogtreecommitdiff
path: root/tests/incite/collections/test_df_collection_thl_marketplaces.py
diff options
context:
space:
mode:
authorMax Nanis2026-09-03 11:20:32 -0700
committerMax Nanis2026-09-03 11:20:32 -0700
commitb2450fa8be202941bb1b3770267fe6b9215d4c27 (patch)
treebee446198b8aa2f6472e99e17765a8f101d5cf2b /tests/incite/collections/test_df_collection_thl_marketplaces.py
parent3c4fdaf7804999bcadc32bc6bb6fce2ad0435a61 (diff)
downloadgeneralresearch-b2450fa8be202941bb1b3770267fe6b9215d4c27.tar.gz
generalresearch-b2450fa8be202941bb1b3770267fe6b9215d4c27.zip
pytest tests -vs RUNS, but obv doesn't all PASS
Diffstat (limited to 'tests/incite/collections/test_df_collection_thl_marketplaces.py')
-rw-r--r--tests/incite/collections/test_df_collection_thl_marketplaces.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/incite/collections/test_df_collection_thl_marketplaces.py b/tests/incite/collections/test_df_collection_thl_marketplaces.py
index 6ad0cb4..0f79b81 100644
--- a/tests/incite/collections/test_df_collection_thl_marketplaces.py
+++ b/tests/incite/collections/test_df_collection_thl_marketplaces.py
@@ -1,3 +1,4 @@
+from collections.abc import Generator
from datetime import UTC, datetime
from itertools import product
from typing import TYPE_CHECKING
@@ -18,8 +19,8 @@ if TYPE_CHECKING:
from generalresearch.pg_helper import PostgresConfig
-def combo_object():
- for x in product(
+def combo_object() -> Generator[tuple[type, str]]:
+ yield from product(
[
InnovateSurveyHistoryCollection,
MorningSurveyTimeseriesCollection,
@@ -27,8 +28,7 @@ def combo_object():
SpectrumSurveyTimeseriesCollection,
],
["5min", "6H", "30D"],
- ):
- yield from x
+ )
@pytest.mark.parametrize("df_coll, offset", combo_object())