]>
git.proxmox.com Git - ceph.git/blob - ceph/src/arrow/c_glib/test/test-buffer-output-stream.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 TestBufferOutputStream
< Test
::Unit::TestCase
19 include Helper
::Buildable
22 buffer
= Arrow
::ResizableBuffer.new(0)
23 output_stream
= Arrow
::BufferOutputStream.new(buffer
)
24 output_stream
.write("Hello")
26 assert_equal("Hello", buffer
.data.to_s
)
30 buffer
= Arrow
::ResizableBuffer.new(0)
31 output_stream
= Arrow
::BufferOutputStream.new(buffer
)
32 output_stream
.write("Hello")
33 output_stream
.align(8)
35 assert_equal("Hello\x00\x00\x00", buffer
.data.to_s
)
38 def test_write_record_batch
40 Arrow
::Field.new("visible", Arrow
::BooleanDataType.new
),
41 Arrow
::Field.new("valid", Arrow
::BooleanDataType.new
),
43 schema
= Arrow
::Schema.new(fields
)
45 build_boolean_array([true]),
46 build_boolean_array([false]),
48 record_batch
= Arrow
::RecordBatch.new(schema
, 1, columns
)
50 buffer
= Arrow
::ResizableBuffer.new(0)
51 options
= Arrow
::WriteOptions.new
52 output_stream
= Arrow
::BufferOutputStream.new(buffer
)
53 output_stream
.write_record_batch(record_batch
, options
)
56 input_stream
= Arrow
::BufferInputStream.new(buffer
)
57 assert_equal(record_batch
,
58 input_stream
.read_record_batch(schema
))