ceph/src/arrow/ruby/red-arrow/test/test-feather.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 FeatherTest < Test::Unit::TestCase
  19   include Helper::Fixture
  20
  21   def setup
  22     columns = {
  23       "message" => Arrow::StringArray.new(["Start", "Crash", "Shutdown"]),
  24       "is_critical" => Arrow::BooleanArray.new([false, true, false]),
  25     }
  26     @table = Arrow::Table.new(columns)
  27
  28     @output = Tempfile.new(["red-arrow", ".feather"])
  29     begin
  30       yield(@output)
  31     ensure
  32       @output.close!
  33     end
  34   end
  35
  36   def test_default
  37     @table.save(@output.path)
  38     @output.close
  39
  40     assert_equal(@table, Arrow::Table.load(@output.path))
  41   end
  42
  43   def test_compression
  44     @table.save(@output.path, compression: :zstd)
  45     @output.close
  46
  47     assert_equal(@table, Arrow::Table.load(@output.path))
  48   end
  49 end