1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
#include "catch.hpp"
#include "test_helpers.hpp"
#include "tpch_extension.hpp"
#include <chrono>
#include <iostream>
#include "duckdb/common/string_util.hpp"
using namespace duckdb;
using namespace std;
TEST_CASE("Test TPC-H SF0.01 using streaming api", "[tpch][.]") {
duckdb::unique_ptr<QueryResult> result;
DuckDB db(nullptr);
Connection con(db);
double sf = 0.01;
if (!db.ExtensionIsLoaded("tpch")) {
return;
}
REQUIRE_NO_FAIL(con.Query("CALL dbgen(sf=" + to_string(sf) + ")"));
for (idx_t tpch_num = 1; tpch_num <= 22; tpch_num++) {
result = con.SendQuery("pragma tpch(" + to_string(tpch_num) + ");");
duckdb::ColumnDataCollection collection(duckdb::Allocator::DefaultAllocator(), result->types);
while (true) {
auto chunk = result->Fetch();
if (chunk) {
collection.Append(*chunk);
} else {
break;
}
}
COMPARE_CSV_COLLECTION(collection, TpchExtension::GetAnswer(sf, tpch_num), true);
}
}
|
