| 170 | end |
| 171 | |
| 172 | def test_add_column |
| 173 | field = Arrow::Field.new("added", Arrow::BooleanDataType.new) |
| 174 | column = build_boolean_array([false, false, true, true, true]) |
| 175 | new_record_batch = @record_batch.add_column(1, field, column) |
| 176 | assert_equal(["visible", "added", "valid"], |
| 177 | new_record_batch.schema.fields.collect(&:name)) |
| 178 | end |
| 179 | |
| 180 | def test_remove_column |
| 181 | new_record_batch = @record_batch.remove_column(0) |