MCPcopy Create free account
hub / github.com/apache/arrow / test_write_table

Method test_write_table

c_glib/test/test-file-writer.rb:59–90  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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")

Callers

nothing calls this directly

Calls 7

openMethod · 0.45
newMethod · 0.45
pathMethod · 0.45
write_tableMethod · 0.45
closeMethod · 0.45
lengthMethod · 0.45
read_record_batchMethod · 0.45

Tested by

no test coverage detected