aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorivanmorozov <ivanmorozov@yandex-team.com>2023-07-13 20:38:38 +0300
committerivanmorozov <ivanmorozov@yandex-team.com>2023-07-13 20:38:38 +0300
commitecb193692e96a585cb6da07242eacbd3639082a1 (patch)
tree005cfb74356a008afa9b4d1e3ffe55f9fdd10932
parent30ac7a6ff269ab84f978aeceae20adedb84f8ec3 (diff)
downloadydb-ecb193692e96a585cb6da07242eacbd3639082a1.tar.gz
KIKIMR-18453: fix tests
-rw-r--r--ydb/core/formats/arrow/ut/ut_size_calcer.cpp2
-rw-r--r--ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp8
2 files changed, 5 insertions, 5 deletions
diff --git a/ydb/core/formats/arrow/ut/ut_size_calcer.cpp b/ydb/core/formats/arrow/ut/ut_size_calcer.cpp
index 879827e124..3aec7d6fbf 100644
--- a/ydb/core/formats/arrow/ut/ut_size_calcer.cpp
+++ b/ydb/core/formats/arrow/ut/ut_size_calcer.cpp
@@ -40,7 +40,7 @@ Y_UNIT_TEST_SUITE(SizeCalcer) {
"field", NConstruction::TStringPoolFiller(1, 0));
std::shared_ptr<arrow::RecordBatch> batch = NConstruction::TRecordBatchConstructor({ column }).BuildBatch(2048);
Cerr << GetBatchDataSize(batch) << Endl;
- UNIT_ASSERT(GetBatchDataSize(batch) == 8 + 2048);
+ UNIT_ASSERT(GetBatchDataSize(batch) == 2048 + 4);
}
Y_UNIT_TEST(SimpleInt64) {
diff --git a/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp b/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
index df84b3fcea..8a7977b478 100644
--- a/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
+++ b/ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp
@@ -468,8 +468,8 @@ Y_UNIT_TEST_SUITE(YdbTableBulkUpsertOlap) {
auto bigBatch = TTestOlap::SampleBatch(true, 130000);
ui32 batchSize = NArrow::GetBatchDataSize(bigBatch);
Cerr << "rows: " << bigBatch->num_rows() << " batch size: " << batchSize << Endl;
- UNIT_ASSERT(batchSize > 15 * 1024 * 1024);
- UNIT_ASSERT(batchSize < 17 * 1024 * 1024);
+ UNIT_ASSERT(batchSize > 15 * 1024 * 1024 + 130000 * 4 * (ui32)bigBatch->num_columns());
+ UNIT_ASSERT(batchSize < 17 * 1024 * 1024 + 130000 * 4 * (ui32)bigBatch->num_columns());
TString bigCsv = TTestOlap::ToCSV(bigBatch);
@@ -484,8 +484,8 @@ Y_UNIT_TEST_SUITE(YdbTableBulkUpsertOlap) {
auto bigBatch = TTestOlap::SampleBatch(true, 150000); // 2 shards, greater then 8 Mb per shard
ui32 batchSize = NArrow::GetBatchDataSize(bigBatch);
Cerr << "rows: " << bigBatch->num_rows() << " batch size: " << batchSize << Endl;
- UNIT_ASSERT(batchSize > 16 * 1024 * 1024);
- UNIT_ASSERT(batchSize < 20 * 1024 * 1024);
+ UNIT_ASSERT(batchSize > 16 * 1024 * 1024 + 150000 * 4 * (ui32)bigBatch->num_columns());
+ UNIT_ASSERT(batchSize < 20 * 1024 * 1024 + 150000 * 4 * (ui32)bigBatch->num_columns());
TString bigCsv = TTestOlap::ToCSV(bigBatch);