ceph/src/arrow/ruby/red-arrow/test/test-record-batch-iterator.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
   8 #
   9 #   http://www.apache.org/licenses/LICENSE-2.0
  10 #
  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
  16 # under the License.
  17
  18 class RecordBatchIteratorTest < Test::Unit::TestCase
  19   def setup
  20     @schema = Arrow::Schema.new(visible: :boolean,
  21                                 count: :uint32)
  22     @record_batches = [
  23       Arrow::RecordBatch.new(@schema,
  24                              visible: [true],
  25                              count: [1]),
  26       Arrow::RecordBatch.new(@schema,
  27                              visible: [false, nil],
  28                              count: [nil, 3]),
  29     ]
  30     @iterator = Arrow::RecordBatchIterator.new(@record_batches)
  31   end
  32
  33   def test_to_a
  34     assert_equal(@record_batches,
  35                  @iterator.to_a)
  36   end
  37 end