Remove unused parameter in TestHashJoinOperator#testMemoryLimit
@@ -648,8 +648,8 @@ public class TestHashJoinOperator
assertOperatorEquals(joinOperatorFactory, taskContext.addPipelineContext(0, true, true).addDriverContext(), probeInput, expected, true, getHashChannels(probePages, buildPages));
}
- @Test(expectedExceptions = ExceededMemoryLimitException.class, expectedExceptionsMessageRegExp = "Query exceeded local memory limit of.*", dataProvider = "hashEnabledValues")
- public void testMemoryLimit(boolean parallelBuild, boolean probeHashEnabled, boolean buildHashEnabled)
+ @Test(expectedExceptions = ExceededMemoryLimitException.class, expectedExceptionsMessageRegExp = "Query exceeded local memory limit of.*", dataProvider = "testMemoryLimitProvider")
+ public void testMemoryLimit(boolean parallelBuild, boolean buildHashEnabled)
throws Exception
{
TaskContext taskContext = TestingTaskContext.createTaskContext(executor, TEST_SESSION, new DataSize(100, BYTE));
@@ -659,6 +659,16 @@ public class TestHashJoinOperator
buildHash(parallelBuild, taskContext, Ints.asList(0), buildPages, Optional.empty());
}
+ @DataProvider
+ public static Object[][] testMemoryLimitProvider()
+ {
+ return new Object[][] {
+ {true, true},
+ {true, false},
+ {false, true},
+ {false, false}};
+ }
+
private TaskContext createTaskContext()
{
return TestingTaskContext.createTaskContext(executor, TEST_SESSION);
