mirror of
https://github.com/apache/arrow.git
synced 2026-10-29 09:05:47 +08:00
ARROW-5362: Fix compression test memory usage
This commit is contained in:
1 parent
6c63c79597
commit
9b51c58091
1 file changed
+7
-5
@@ -69,17 +69,16 @@ void CheckCodecRoundtrip(Compression::type ctype, const std::vector<uint8_t>& da
|
||||
int64_t actual_size;
|
||||
ASSERT_OK(c1->Compress(data.size(), data.data(), max_compressed_len, compressed.data(),
|
||||
&actual_size));
|
||||
compressed.resize(actual_size);
|
||||
|
||||
// decompress with c2
|
||||
ASSERT_OK(c2->Decompress(compressed.size(), compressed.data(), decompressed.size(),
|
||||
ASSERT_OK(c2->Decompress(actual_size, compressed.data(), decompressed.size(),
|
||||
decompressed.data()));
|
||||
|
||||
ASSERT_EQ(data, decompressed);
|
||||
|
||||
// decompress with size with c2
|
||||
int64_t actual_decompressed_size;
|
||||
ASSERT_OK(c2->Decompress(compressed.size(), compressed.data(), decompressed.size(),
|
||||
ASSERT_OK(c2->Decompress(actual_size, compressed.data(), decompressed.size(),
|
||||
decompressed.data(), &actual_decompressed_size));
|
||||
|
||||
ASSERT_EQ(data, decompressed);
|
||||
@@ -87,19 +86,22 @@ void CheckCodecRoundtrip(Compression::type ctype, const std::vector<uint8_t>& da
|
||||
|
||||
// compress with c2
|
||||
int64_t actual_size2;
|
||||
ASSERT_EQ(max_compressed_len,
|
||||
static_cast<int>(c2->MaxCompressedLen(data.size(), data.data())));
|
||||
compressed.resize(max_compressed_len);
|
||||
ASSERT_OK(c2->Compress(data.size(), data.data(), max_compressed_len, compressed.data(),
|
||||
&actual_size2));
|
||||
ASSERT_EQ(actual_size2, actual_size);
|
||||
|
||||
// decompress with c1
|
||||
ASSERT_OK(c1->Decompress(compressed.size(), compressed.data(), decompressed.size(),
|
||||
ASSERT_OK(c1->Decompress(actual_size2, compressed.data(), decompressed.size(),
|
||||
decompressed.data()));
|
||||
|
||||
ASSERT_EQ(data, decompressed);
|
||||
|
||||
// decompress with size with c1
|
||||
int64_t actual_decompressed_size2;
|
||||
ASSERT_OK(c1->Decompress(compressed.size(), compressed.data(), decompressed.size(),
|
||||
ASSERT_OK(c1->Decompress(actual_size2, compressed.data(), decompressed.size(),
|
||||
decompressed.data(), &actual_decompressed_size2));
|
||||
|
||||
ASSERT_EQ(data, decompressed);
|
||||
|
||||
Reference in new issue
Block a user