| 100 | } // namespace |
| 101 | |
| 102 | Result<std::shared_ptr<Buffer>> MakeSampleCsvBuffer( |
| 103 | size_t num_rows, std::function<bool(size_t)> is_valid) { |
| 104 | std::stringstream writer; |
| 105 | |
| 106 | WriteHeader(writer); |
| 107 | for (size_t i = 0; i < num_rows; ++i) { |
| 108 | if (!is_valid || is_valid(i)) { |
| 109 | WriteRow(writer, i); |
| 110 | } else { |
| 111 | WriteInvalidRow(writer, i); |
| 112 | } |
| 113 | } |
| 114 | |
| 115 | auto table_str = writer.str(); |
| 116 | auto table_buffer = std::make_shared<Buffer>(table_str); |
| 117 | return MemoryManager::CopyBuffer(table_buffer, default_cpu_memory_manager()); |
| 118 | } |
| 119 | |
| 120 | } // namespace csv |
| 121 | } // namespace arrow |
nothing calls this directly
no test coverage detected