Skip to content

Commit 79cac45

Browse files
authored
Correct FRACTION substitution parameter for Q11 (#163)
1 parent 14219f4 commit 79cac45

File tree

3 files changed

+18
-4
lines changed

3 files changed

+18
-4
lines changed

queries/duckdb/q11.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
import duckdb
22

33
from queries.duckdb import utils
4+
from settings import Settings
5+
6+
settings = Settings()
47

58
Q_NUM = 11
69

@@ -9,6 +12,8 @@ def q() -> None:
912
supplier_ds = utils.get_supplier_ds()
1013
part_supp_ds = utils.get_part_supp_ds()
1114
nation_ds = utils.get_nation_ds()
15+
scale_factor = settings.scale_factor
16+
fraction = 0.0001 / scale_factor
1217

1318
query_str = f"""
1419
select
@@ -26,7 +31,7 @@ def q() -> None:
2631
ps_partkey having
2732
sum(ps_supplycost * ps_availqty) > (
2833
select
29-
sum(ps_supplycost * ps_availqty) * 0.0001
34+
sum(ps_supplycost * ps_availqty) * {fraction}
3035
from
3136
{part_supp_ds},
3237
{supplier_ds},

queries/polars/q11.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,9 @@
33
import polars as pl
44

55
from queries.polars import utils
6+
from settings import Settings
7+
8+
settings = Settings()
69

710
Q_NUM = 11
811

@@ -23,7 +26,7 @@ def q(
2326
assert supplier is not None
2427

2528
var1 = "GERMANY"
26-
var2 = 0.0001
29+
var2 = 0.0001 / settings.scale_factor
2730

2831
q1 = (
2932
partsupp.join(supplier, left_on="ps_suppkey", right_on="s_suppkey")

queries/pyspark/q11.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,16 @@
11
from queries.pyspark import utils
2+
from settings import Settings
3+
4+
settings = Settings()
25

36
Q_NUM = 11
47

58

69
def q() -> None:
7-
query_str = """
10+
scale_factor = settings.scale_factor
11+
fraction = 0.0001 / scale_factor
12+
13+
query_str = f"""
814
select
915
ps_partkey,
1016
round(sum(ps_supplycost * ps_availqty), 2) as value
@@ -20,7 +26,7 @@ def q() -> None:
2026
ps_partkey having
2127
sum(ps_supplycost * ps_availqty) > (
2228
select
23-
sum(ps_supplycost * ps_availqty) * 0.0001
29+
sum(ps_supplycost * ps_availqty) * {fraction}
2430
from
2531
partsupp,
2632
supplier,

0 commit comments

Comments
 (0)