From 56a0ab1ee51843a038fb796195c1c3ed8e4c3c36 Mon Sep 17 00:00:00 2001 From: Dan Corneanu Date: Sun, 19 Jan 2025 15:10:19 +1300 Subject: [PATCH] Remove redundant assert --- lib/red_amber/data_frame_variable_operation.rb | 6 +++++- test/test_data_frame_variable_operation.rb | 1 - 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/lib/red_amber/data_frame_variable_operation.rb b/lib/red_amber/data_frame_variable_operation.rb index 40b4ad16..a1681c15 100755 --- a/lib/red_amber/data_frame_variable_operation.rb +++ b/lib/red_amber/data_frame_variable_operation.rb @@ -694,7 +694,11 @@ def append_to_fields_and_arrays(appender, fields, arrays, append_to_left) raise DataFrameArgumentError, "Data size mismatch (#{data.size} != #{size})" end - a = Arrow::Array.new(data.is_a?(Vector) ? data.to_a : data) + if data.is_a?(Vector) + a = data.data + else + a = Arrow::Array.new(data) + end if append_to_left fields.unshift(Arrow::Field.new(key.to_sym, a.value_data_type)) diff --git a/test/test_data_frame_variable_operation.rb b/test/test_data_frame_variable_operation.rb index ebb0c43c..26a65c8d 100755 --- a/test/test_data_frame_variable_operation.rb +++ b/test/test_data_frame_variable_operation.rb @@ -441,7 +441,6 @@ class DataFrameVariableOperationTest < Test::Unit::TestCase sub_test_case 'Dataframe with zero n_records' do test 'assign by block' do - assert_equal :double, @df.b.type str = <<~STR RedAmber::DataFrame : 0 x 4 Vectors Vectors : 2 numeric, 1 string, 1 boolean