diff --git a/ruby/red-arrow-format/lib/arrow-format/type.rb b/ruby/red-arrow-format/lib/arrow-format/type.rb index 6711f995fdb..1124a13f6a8 100644 --- a/ruby/red-arrow-format/lib/arrow-format/type.rb +++ b/ruby/red-arrow-format/lib/arrow-format/type.rb @@ -548,6 +548,23 @@ def build_array(...) end class TimestampType < TemporalType + class << self + def try_convert(value) + case value + when Symbol + unit = value + new(unit, nil) + when ::Array + unit, time_zone = value + new(unit, time_zone) + when self + value + else + nil + end + end + end + attr_reader :unit attr_reader :time_zone def initialize(unit, time_zone) @@ -556,6 +573,12 @@ def initialize(unit, time_zone) @time_zone = time_zone end + def ==(other) + other.is_a?(self.class) and + @unit == other.unit and + @time_zone == other.time_zone + end + def name "Timestamp" end @@ -564,6 +587,10 @@ def buffer_type :s64 end + def pack_template + "q" + end + def build_array(...) TimestampArray.new(self, ...) end diff --git a/ruby/red-arrow-format/test/test-timestamp-array.rb b/ruby/red-arrow-format/test/test-timestamp-array.rb new file mode 100644 index 00000000000..397a01a64f0 --- /dev/null +++ b/ruby/red-arrow-format/test/test-timestamp-array.rb @@ -0,0 +1,86 @@ +# 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 TestTimestampArray < Test::Unit::TestCase + def setup + @timestamp_2019_11_17_15_09_11 = 1574003351 + @timestamp_2025_12_16_05_33_58 = 1765863238 + end + + sub_test_case("#initialize") do + def test_no_null + values = [ + @timestamp_2019_11_17_15_09_11, + @timestamp_2025_12_16_05_33_58, + ] + assert_equal(values, + ArrowFormat::TimestampArray.new(:second, values).to_a) + end + + def test_mixed + values = [ + @timestamp_2019_11_17_15_09_11, + nil, + @timestamp_2025_12_16_05_33_58, + ] + assert_equal(values, + ArrowFormat::TimestampArray.new(:second, values).to_a) + end + + def test_time_zone + time_zone = "UTC" + array = ArrowFormat::TimestampArray.new([:second, time_zone], [nil]) + assert_equal(ArrowFormat::TimestampType.new(:second, time_zone), + array.type) + end + end + + sub_test_case("#==") do + def test_no_slice + values = [ + @timestamp_2019_11_17_15_09_11, + nil, + @timestamp_2025_12_16_05_33_58, + ] + array1 = ArrowFormat::TimestampArray.new(:second, values) + array2 = ArrowFormat::TimestampArray.new(:second, values) + assert_equal(array1, array2) + end + + def test_sliced + values = [ + @timestamp_2019_11_17_15_09_11, + nil, + @timestamp_2025_12_16_05_33_58, + ] + array1 = ArrowFormat::TimestampArray.new(:second, values) + array2 = ArrowFormat::TimestampArray.new(:second, [0, *values, 0]) + assert_equal(array1, array2.slice(1, 3)) + end + + def test_sliced_different_content + values = [ + @timestamp_2019_11_17_15_09_11, + nil, + @timestamp_2025_12_16_05_33_58, + ] + array1 = ArrowFormat::TimestampArray.new(:second, values) + array2 = ArrowFormat::TimestampArray.new(:second, [0, 0, *values, 0]) + assert_not_equal(array1, array2.slice(1, 3)) + end + end +end