diff options
| author | ivanmorozov <ivanmorozov@yandex-team.com> | 2023-07-13 20:38:38 +0300 |
|---|---|---|
| committer | ivanmorozov <ivanmorozov@yandex-team.com> | 2023-07-13 20:38:38 +0300 |
| commit | ecb193692e96a585cb6da07242eacbd3639082a1 (patch) | |
| tree | 005cfb74356a008afa9b4d1e3ffe55f9fdd10932 | |
| parent | 30ac7a6ff269ab84f978aeceae20adedb84f8ec3 (diff) | |
| download | ydb-ecb193692e96a585cb6da07242eacbd3639082a1.tar.gz | |
KIKIMR-18453: fix tests
| -rw-r--r-- | ydb/core/formats/arrow/ut/ut_size_calcer.cpp | 2 | ||||
| -rw-r--r-- | ydb/services/ydb/ydb_bulk_upsert_olap_ut.cpp | 8 |
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); |