From 02b5eca655c88f298ca9ced72a35dd07002fd51d Mon Sep 17 00:00:00 2001 From: stanbrub Date: Tue, 5 Nov 2024 21:31:40 -0700 Subject: [PATCH] Added partition_by test --- .../tests/standard/by/PartitionByTest.java | 32 ++++++++++--------- 1 file changed, 17 insertions(+), 15 deletions(-) diff --git a/src/it/java/io/deephaven/benchmark/tests/standard/by/PartitionByTest.java b/src/it/java/io/deephaven/benchmark/tests/standard/by/PartitionByTest.java index ae7ff946..ee699aee 100644 --- a/src/it/java/io/deephaven/benchmark/tests/standard/by/PartitionByTest.java +++ b/src/it/java/io/deephaven/benchmark/tests/standard/by/PartitionByTest.java @@ -7,7 +7,6 @@ /** * Standard tests for the partitionBy table operation. Divides a single table into subtables. */ -@Disabled public class PartitionByTest { final StandardTestRunner runner = new StandardTestRunner(this); @@ -17,21 +16,24 @@ void setup() { } @Test - void partitionBy1Group2Cols() { - var q = "source.partition_by(['key1']).get_constituent(['string1val'])"; - runner.test("PartitionBy- 1 Group 100 Unique Vals", 1, q, "key1", "num1"); + void partitionBy1Group() { + runner.setScaleFactors(44, 9); + var q = "source.partition_by(by=['key1']).table"; + runner.test("PartitionBy- 1 Group 100 Unique Vals", 100, q, "key1", "num1"); } - // @Test - // void partitionBy1Group2ColsLarge() { - // var q = "source.partition_by(['str3']).get_constituent(['val1string'])"; - // runner.test("PartitionBy- 1 Group 1M Unique Vals", 1000000, q, "str3", "int1M"); - // } - // - // @Test - // public void partitionBy2Group3Cols() { - // var q = "source.partition_by(by=['str1', 'str2'])"; - // runner.test("PartitionBy- 2 Group 160K Unique Combos", 160000, q, "str1", "str2", "int250"); - // } + @Test + void partitionBy2Groups() { + runner.setScaleFactors(8, 4); + var q = "source.partition_by(by=['key1', 'key2']).table"; + runner.test("PartitionBy- 2 Groups 10K Unique Combos", 10100, q, "key1", "key2", "num1"); + } + + @Test + void partitionBy3Groups() { + runner.setScaleFactors(4, 2); + var q = "source.partition_by(by=['key1', 'key2', 'key3']).table"; + runner.test("PartitionBy- 3 Groups 100K Unique Combos", 90900, q, "key1", "key2", "key3", "num1"); + } }