| 57 | end |
| 58 | |
| 59 | def test_write_table |
| 60 | tempfile = Tempfile.open("arrow-ipc-file-writer") |
| 61 | output = Arrow::FileOutputStream.new(tempfile.path, false) |
| 62 | |
| 63 | array = build_boolean_array([true, false, true]) |
| 64 | field = Arrow::Field.new("enabled", Arrow::BooleanDataType.new) |
| 65 | schema = Arrow::Schema.new([field]) |
| 66 | |
| 67 | begin |
| 68 | file_writer = Arrow::RecordBatchFileWriter.new(output, schema) |
| 69 | begin |
| 70 | table = Arrow::Table.new(schema, [array]) |
| 71 | file_writer.write_table(table) |
| 72 | ensure |
| 73 | file_writer.close |
| 74 | assert do |
| 75 | file_writer.closed? |
| 76 | end |
| 77 | end |
| 78 | ensure |
| 79 | output.close |
| 80 | end |
| 81 | |
| 82 | input = Arrow::MemoryMappedInputStream.new(tempfile.path) |
| 83 | begin |
| 84 | file_reader = Arrow::RecordBatchFileReader.new(input) |
| 85 | assert_equal(Arrow::RecordBatch.new(schema, array.length, [array]), |
| 86 | file_reader.read_record_batch(0)) |
| 87 | ensure |
| 88 | input.close |
| 89 | end |
| 90 | end |
| 91 | |
| 92 | def test_footer_custom_metadata |
| 93 | tempfile = Tempfile.open("arrow-ipc-file-writer") |
nothing calls this directly
no test coverage detected