Class TableTest

ruby/red-arrow/test/test-table.rb:18–1534

# under the License.

class TableTest < Test::Unit::TestCase
  include Helper::Fixture

  def setup
    @count_field = Arrow::Field.new("count", :uint8)
    @visible_field = Arrow::Field.new("visible", :boolean)
    schema = Arrow::Schema.new([@count_field, @visible_field])
    count_arrays = [
      Arrow::UInt8Array.new([1, 2]),
      Arrow::UInt8Array.new([4, 8, 16]),
      Arrow::UInt8Array.new([32, 64]),
      Arrow::UInt8Array.new([128]),
    ]
    visible_arrays = [
      Arrow::BooleanArray.new([true, false, nil]),
      Arrow::BooleanArray.new([true]),
      Arrow::BooleanArray.new([true, false]),
      Arrow::BooleanArray.new([nil]),
      Arrow::BooleanArray.new([nil]),
    ]
    @count_array = Arrow::ChunkedArray.new(count_arrays)
    @visible_array = Arrow::ChunkedArray.new(visible_arrays)
    @table = Arrow::Table.new(schema, [@count_array, @visible_array])
  end

  sub_test_case(".new") do
    test("{Symbol: Arrow::Array}") do
      schema = Arrow::Schema.new(numbers: :int64)
      assert_equal(Arrow::Table.new(schema,
                                    [Arrow::Int64Array.new([1, 2, 3])]),
                   Arrow::Table.new(numbers: Arrow::Int64Array.new([1, 2, 3])))
    end

    test("{Symbol: Arrow::ChunkedArray}") do
      chunked_array = Arrow::ChunkedArray.new([Arrow::Int64Array.new([1, 2, 3])])
      schema = Arrow::Schema.new(numbers: :int64)
      assert_equal(Arrow::Table.new(schema,
                                    [Arrow::Int64Array.new([1, 2, 3])]),
                   Arrow::Table.new(numbers: chunked_array))
    end

    test("{Symbol: Arrow::Tensor}") do
      schema = Arrow::Schema.new(numbers: :uint8)
      assert_equal(Arrow::Table.new(schema,
                                    [Arrow::UInt8Array.new([1, 2, 3])]),
                   Arrow::Table.new(numbers: Arrow::Tensor.new([1, 2, 3])))
    end

    test("{Symbol: #to_ary}") do
      array_like = Object.new
      def array_like.to_ary
        [1, 2, 3]
      end
      schema = Arrow::Schema.new(numbers: :uint8)
      assert_equal(Arrow::Table.new(schema, [Arrow::UInt8Array.new([1, 2, 3])]),
                   Arrow::Table.new(numbers: array_like))
    end
  end