public StatsCalculatorTester() { this(testSessionBuilder().build()); }
public static LocalQueryRunner createLocalQueryRunner(Map<String, String> extraSessionProperties) { SessionBuilder sessionBuilder = testSessionBuilder() .setCatalog("tpch") .setSchema(TINY_SCHEMA_NAME); extraSessionProperties.forEach(sessionBuilder::setSystemProperty); Session session = sessionBuilder.build(); LocalQueryRunner localQueryRunner = new LocalQueryRunner(session); // add tpch localQueryRunner.createCatalog("tpch", new TpchConnectorFactory(1), ImmutableMap.of()); return localQueryRunner; } }
@BeforeClass public void setUp() { queryRunner = new LocalQueryRunner(testSessionBuilder().build()); }
@BeforeClass public void setUp() { queryRunner = new LocalQueryRunner(testSessionBuilder().build()); }
@Test public void testDeallocate() { Session session = testSessionBuilder() .addPreparedStatement("my_query", "SELECT bar, baz FROM foo") .build(); Set<String> statements = executeDeallocate("my_query", "DEALLOCATE PREPARE my_query", session); assertEquals(statements, ImmutableSet.of("my_query")); }
private static SessionBuilder sessionBuilder() { return testSessionBuilder() .setCatalog("tpch") .setSchema(TINY_SCHEMA_NAME); } }
private static SessionBuilder sessionBuilder() { return testSessionBuilder() .setCatalog("tpch") .setSchema(TINY_SCHEMA_NAME); } }
private static SessionBuilder sessionBuilder() { return testSessionBuilder() .setCatalog("tpch") .setSchema(TINY_SCHEMA_NAME); } }
public static void main(String[] args) { new HashBuildAndJoinBenchmark(testSessionBuilder().build(), createLocalQueryRunner()).runBenchmark(new SimpleLineBenchmarkResultWriter(System.out)); new HashBuildAndJoinBenchmark(testSessionBuilder().build(), createLocalQueryRunnerHashEnabled()).runBenchmark(new SimpleLineBenchmarkResultWriter(System.out)); } }
protected TestDateTimeFunctionsBase(boolean legacyTimestamp) { super(testSessionBuilder() .setSystemProperty("legacy_timestamp", String.valueOf(legacyTimestamp)) .setTimeZoneKey(TIME_ZONE_KEY) .setStartTime(new DateTime(2017, 4, 1, 12, 34, 56, 789, UTC_TIME_ZONE).getMillis()) .build()); TIMESTAMP = legacyTimestamp ? LEGACY_TIMESTAMP : NEW_TIMESTAMP; }
@BeforeClass public void setUp() { calculator = new ScalarStatsCalculator(MetadataManager.createTestMetadataManager()); session = testSessionBuilder().build(); }
public TestTimeBase(boolean legacyTimestamp) { super(testSessionBuilder() .setSystemProperty("legacy_timestamp", String.valueOf(legacyTimestamp)) .setTimeZoneKey(TIME_ZONE_KEY) .build()); }
protected TestDateTimeOperatorsBase(boolean legacyTimestamp) { super(testSessionBuilder() .setSystemProperty("legacy_timestamp", String.valueOf(legacyTimestamp)) .setTimeZoneKey(TIME_ZONE_KEY) .build()); }
protected TestTimestampBase(boolean legacyTimestamp) { super(testSessionBuilder() .setSystemProperty("legacy_timestamp", String.valueOf(legacyTimestamp)) .setTimeZoneKey(TIME_ZONE_KEY) .build()); }
protected TestTimestampWithTimeZoneBase(boolean legacyTimestamp) { super(testSessionBuilder() .setSystemProperty("legacy_timestamp", String.valueOf(legacyTimestamp)) .setTimeZoneKey(TIME_ZONE_KEY) .build()); }
@BeforeClass public void init() { Session session = testSessionBuilder() .setSystemProperty(OPTIMIZE_HASH_GENERATION, "true") .build(); assertions = new QueryAssertions(session); }
@BeforeClass @Override public void init() { assertions = new QueryAssertions(testSessionBuilder() .setSystemProperty(USE_MARK_DISTINCT, "false") .build()); } }
@BeforeClass public void setupClass() { defaultFilterTester = new StatsCalculatorTester( testSessionBuilder() .setSystemProperty("default_filter_factor_enabled", "true") .build()); }
protected TestDateBase(boolean legacyTimestamp) { super(testSessionBuilder() .setSystemProperty("legacy_timestamp", String.valueOf(legacyTimestamp)) .setTimeZoneKey(TIME_ZONE_KEY) .build()); }
@Setup public void setUp() throws IOException { queryRunner = new LocalQueryRunner(testSessionBuilder() .setCatalog("memory") .setSchema("default") .build()); queryRunner.installPlugin(new GeoPlugin()); queryRunner.createCatalog("memory", new MemoryConnectorFactory(), ImmutableMap.of()); Path path = Paths.get(BenchmarkSpatialJoin.class.getClassLoader().getResource("us-states.tsv").getPath()); String polygonValues = Files.lines(path) .map(line -> line.split("\t")) .map(parts -> format("('%s', '%s')", parts[0], parts[1])) .collect(Collectors.joining(",")); queryRunner.execute(format("CREATE TABLE memory.default.polygons AS SELECT * FROM (VALUES %s) as t (name, wkt)", polygonValues)); }