LouisClt commented on code in PR #14219:
URL: https://github.com/apache/arrow/pull/14219#discussion_r983158783
##########
cpp/src/arrow/adapters/orc/adapter_test.cc:
##########
@@ -730,4 +735,55 @@ TEST_F(TestORCWriterSingleArray, WriteListOfMap) {
AssertArrayWriteReadEqual(array, array, kDefaultSmallMemStreamSize * 10);
}
+class TestORCWriterMultipleWrite : public ::testing::Test {
+ public:
+ TestORCWriterMultipleWrite() : rand(kRandomSeed) {}
+
+ protected:
+ random::RandomArrayGenerator rand;
+};
+
+TEST_F(TestORCWriterMultipleWrite, MultipleWritesIntField) {
+ const int64_t num_rows = 1234;
+ const int nb_writes = 5;
+ auto array_int = rand.ArrayOf(int32(), num_rows, 0);
+ std::shared_ptr<Schema> input_schema = schema({field("col0",
array_int->type())});
+ ArrayVector vect;
+ for (int i = 0; i < nb_writes; i++) {
+ vect.push_back(array_int);
+ }
+ auto input_chunked_array = std::make_shared<ChunkedArray>(array_int),
+ expected_output_chunked_array = std::make_shared<ChunkedArray>(vect);
+ std::shared_ptr<Table> input_table = Table::Make(input_schema,
{input_chunked_array}),
+ expected_output_table =
+ Table::Make(input_schema,
{expected_output_chunked_array});
+ AssertTableWriteReadEqual(input_table, expected_output_table,
+ kDefaultSmallMemStreamSize * 100, nb_writes);
+}
+
+TEST_F(TestORCWriterMultipleWrite, MultipleWritesIncoherentSchema) {
+ const int64_t num_rows = 1234;
+ auto array_int = rand.ArrayOf(int32(), num_rows, 0);
+ std::shared_ptr<Schema> input_schema = schema({field("col0",
array_int->type())});
+ auto array_int2 = rand.ArrayOf(int64(), num_rows, 0);
+ std::shared_ptr<Schema> input_schema2 = schema({field("col0",
array_int2->type())});
+
+ auto input_chunked_array = std::make_shared<ChunkedArray>(array_int),
+ input_chunked_array2 = std::make_shared<ChunkedArray>(array_int2);
+ std::shared_ptr<Table> input_table = Table::Make(input_schema,
{input_chunked_array}),
+ input_table2 =
+ Table::Make(input_schema2,
{input_chunked_array2});
Review Comment:
Ok, thanks. I adapted the code from the "AssertArrayWriteReadEqual" method,
so I kept a good part of the syntax
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
