ARROW-5362: Fix compression test memory usage

This commit is contained in:
Micah Kornfield committed 2019-05-17 10:40:54 -07:00
1 parent 6c63c79597
commit 9b51c58091
1 file changed
+7 -5
+7 -5
View File
@@ -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);