Skip to content

Commit 6651574

Browse files
feldera-botryzhyk
authored andcommitted
[ci] apply automatic fixes
Signed-off-by: feldera-bot <feldera-bot@feldera.com>
1 parent de66890 commit 6651574

File tree

2 files changed

+31
-7
lines changed

2 files changed

+31
-7
lines changed

python/feldera/testutils.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,11 @@ def number_of_processed_records(pipeline: Pipeline) -> int:
347347

348348

349349
def run_workload(
350-
pipeline_name: str, tables: dict, views: List[ViewSpec], transaction: bool = True, stop: bool = True
350+
pipeline_name: str,
351+
tables: dict,
352+
views: List[ViewSpec],
353+
transaction: bool = True,
354+
stop: bool = True,
351355
) -> Pipeline:
352356
"""
353357
Helper to run a pipeline to completion and validate the views afterwards using ad-hoc queries.

python/tests/workloads/test_tpch.py

Lines changed: 26 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1429,19 +1429,39 @@ def tpch_test(config: TPCHTestConfig):
14291429

14301430
elif config.mode == "transaction":
14311431
pipeline = run_workload(
1432-
unique_pipeline_name("tpc-h-transaction"), tables, views, transaction=True, stop=False
1432+
unique_pipeline_name("tpc-h-transaction"),
1433+
tables,
1434+
views,
1435+
transaction=True,
1436+
stop=False,
14331437
)
1434-
count = next(pipeline.query("SELECT COUNT(*) as cnt FROM REGION WHERE R_COMMENT IS NOT NULL"))["cnt"]
1438+
count = next(
1439+
pipeline.query(
1440+
"SELECT COUNT(*) as cnt FROM REGION WHERE R_COMMENT IS NOT NULL"
1441+
)
1442+
)["cnt"]
14351443
if count != 0:
1436-
raise RuntimeError(f"REGION.R_COMMENT is not NULL despite 'skip_unused_columns' = 'true': {count}")
1444+
raise RuntimeError(
1445+
f"REGION.R_COMMENT is not NULL despite 'skip_unused_columns' = 'true': {count}"
1446+
)
14371447
pipeline.stop(force=True)
14381448
elif config.mode == "stream":
14391449
pipeline = run_workload(
1440-
unique_pipeline_name("tpc-h-stream"), tables, views, transaction=False, stop=False
1450+
unique_pipeline_name("tpc-h-stream"),
1451+
tables,
1452+
views,
1453+
transaction=False,
1454+
stop=False,
14411455
)
1442-
count = next(pipeline.query("SELECT COUNT(*) as cnt FROM REGION WHERE R_COMMENT IS NOT NULL"))["cnt"]
1456+
count = next(
1457+
pipeline.query(
1458+
"SELECT COUNT(*) as cnt FROM REGION WHERE R_COMMENT IS NOT NULL"
1459+
)
1460+
)["cnt"]
14431461
if count != 0:
1444-
raise RuntimeError(f"REGION.R_COMMENT is not NULL despite 'skip_unused_columns' = 'true': {count}")
1462+
raise RuntimeError(
1463+
f"REGION.R_COMMENT is not NULL despite 'skip_unused_columns' = 'true': {count}"
1464+
)
14451465
pipeline.stop(force=True)
14461466
else:
14471467
raise RuntimeError(f"Unknown mode: {config.mode}")

0 commit comments

Comments
 (0)