# Licensed to the Apache Software Foundation (ASF) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information # regarding copyright ownership. The ASF licenses this file # to you under the Apache License, Version 2.0 (the # "License"); you may not use this file except in compliance # with the License. You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY # KIND, either express or implied. See the License for the # specific language governing permissions and limitations # under the License. class TestGIOOutputStream < Test::Unit::TestCase include Helper::Buildable def test_writer_backend data = [true] field = Arrow::Field.new("enabled", Arrow::BooleanDataType.new) schema = Arrow::Schema.new([field]) tempfile = Tempfile.open("arrow-gio-output-stream") file = Gio::File.new_for_path(tempfile.path) output_stream = file.append_to(:none) output = Arrow::GIOOutputStream.new(output_stream) begin file_writer = Arrow::RecordBatchFileWriter.new(output, schema) begin record_batch = Arrow::RecordBatch.new(schema, data.size, [build_boolean_array(data)]) file_writer.write_record_batch(record_batch) ensure file_writer.close end ensure output.close end input = Arrow::MemoryMappedInputStream.new(tempfile.path) begin file_reader = Arrow::RecordBatchFileReader.new(input) assert_equal([field.name], file_reader.schema.fields.collect(&:name)) assert_equal(Arrow::RecordBatch.new(schema, data.size, [build_boolean_array(data)]), file_reader.read_record_batch(0)) ensure input.close end end def test_getter output_stream = Gio::MemoryOutputStream.new output = Arrow::GIOOutputStream.new(output_stream) assert_equal(output_stream, output.raw) end def test_tell unless Gio.const_defined?(:UnixOutputStream) omit("Need Gio::UnixOutputStream") end tempfile = Tempfile.open("arrow-gio-output-stream") begin output_stream = Gio::UnixOutputStream.new(tempfile.to_i, false) output = Arrow::GIOOutputStream.new(output_stream) assert_equal(0, output.tell) output.write("Hello") assert_equal(5, output.tell) ensure tempfile.close! end end end