should be it
This commit is contained in:
39
external/duckdb/test/api/test_tpch_with_streaming.cpp
vendored
Normal file
39
external/duckdb/test/api/test_tpch_with_streaming.cpp
vendored
Normal file
@@ -0,0 +1,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);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user