]>
git.proxmox.com Git - ceph.git/blob - ceph/src/arrow/c_glib/test/test-stream-writer.rb
1 # Licensed to the Apache Software Foundation (ASF) under one
2 # or more contributor license agreements. See the NOTICE file
3 # distributed with this work for additional information
4 # regarding copyright ownership. The ASF licenses this file
5 # to you under the Apache License, Version 2.0 (the
6 # "License"); you may not use this file except in compliance
7 # with the License. You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing,
12 # software distributed under the License is distributed on an
13 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14 # KIND, either express or implied. See the License for the
15 # specific language governing permissions and limitations
18 class TestStreamWriter
< Test
::Unit::TestCase
19 include Helper
::Buildable
21 def test_write_record_batch
23 field
= Arrow
::Field.new("enabled", Arrow
::BooleanDataType.new
)
24 schema
= Arrow
::Schema.new([field
])
26 tempfile
= Tempfile
.open("arrow-ipc-stream-writer")
27 output
= Arrow
::FileOutputStream.new(tempfile
.path
, false)
29 stream_writer
= Arrow
::RecordBatchStreamWriter.new(output
, schema
)
32 build_boolean_array(data),
34 record_batch
= Arrow
::RecordBatch.new(schema
, data.size
, columns
)
35 stream_writer
.write_record_batch(record_batch
)
43 input
= Arrow
::MemoryMappedInputStream.new(tempfile
.path
)
45 stream_reader
= Arrow
::RecordBatchStreamReader.new(input
)
46 assert_equal([field
.name
],
47 stream_reader
.schema
.fields
.collect(&:name))
48 assert_equal(Arrow
::RecordBatch.new(schema
,
50 [build_boolean_array(data)]),
51 stream_reader
.read_next
)
52 assert_nil(stream_reader
.read_next
)