Skip to content

Commit 324ef4e

Browse files
authored
test: DH-21118: Added with_serial benchmarks (#415)
1 parent 73d6700 commit 324ef4e

File tree

1 file changed

+41
-0
lines changed

1 file changed

+41
-0
lines changed

src/it/java/io/deephaven/benchmark/tests/standard/formula/UserFormulaTest.java

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,20 @@ def f(num1, num2):
5757
var q = "source.select(['num1=f(num1, num2)','num1=(double)num1'])";
5858
runner.test("UDF- 2 Doubles to Double No Hints", q, "num1", "num2");
5959
}
60+
61+
@Test
62+
void udf2DoublesToDoubleNoHintsSerial() {
63+
var setup = """
64+
def f(num1, num2):
65+
return num1 + num2
66+
from deephaven.table import Selectable
67+
col1 = Selectable.parse('num1=f(num1, num2)').with_serial()
68+
col2 = Selectable.parse('num1=(double)num1').with_serial()
69+
""";
70+
runner.addSetupQuery(setup);
71+
var q = "source.select([col1, col2])";
72+
runner.test("UDF- 2 Doubles to Double No Hints Serial", q, "num1", "num2");
73+
}
6074

6175
@Test
6276
void udf2DoublesToDoubleNoHintsNoVectorize() {
@@ -80,6 +94,20 @@ def f(num1: float, num2: float) -> float:
8094
var q = "source.select(['num1=f(num1, num2)'])";
8195
runner.test("UDF- 2 Doubles to Double Python Hints", q, "num1", "num2");
8296
}
97+
98+
@Test
99+
void udf2DoublesToDoublePythonHintsSerial() {
100+
runner.setScaleFactors(3, 0);
101+
var setup = """
102+
def f(num1: float, num2: float) -> float:
103+
return num1 + num2
104+
from deephaven.table import Selectable
105+
col = Selectable.parse('num1=f(num1, num2)').with_serial()
106+
""";
107+
runner.addSetupQuery(setup);
108+
var q = "source.select([col])";
109+
runner.test("UDF- 2 Doubles to Double Python Hints Serial", q, "num1", "num2");
110+
}
83111

84112
@Test
85113
void udf2DoublesToDoublePythonHintsNoVectorize() {
@@ -126,6 +154,19 @@ def f(num1: np.float64, num2: np.float64) -> np.float64:
126154
var q = "source.select(['num1=f(num1, num2)'])";
127155
runner.test("UDF- 2 Doubles to Double Numpy Hints", q, "num1", "num2");
128156
}
157+
158+
@Test
159+
void udf2DoublesToDoubleNumpyHintsSerial() {
160+
var setup = """
161+
def f(num1: np.float64, num2: np.float64) -> np.float64:
162+
return num1 + num2
163+
from deephaven.table import Selectable
164+
col = Selectable.parse('num1=f(num1, num2)').with_serial()
165+
""";
166+
runner.addSetupQuery(setup);
167+
var q = "source.select([col])";
168+
runner.test("UDF- 2 Doubles to Double Numpy Hints Serial", q, "num1", "num2");
169+
}
129170

130171
@Test
131172
void udf2DoublesToDoubleNumpyHintsNoVectorize() {

0 commit comments

Comments
 (0)