test/test-arrow-table.rb in red-arrow-dataset-10.0.1 vs test/test-arrow-table.rb in red-arrow-dataset-11.0.0
- old
+ new
@@ -74,7 +74,20 @@
assert_equal(Arrow::Table.new(visible: [true, true, true],
point: [1, 3, 10]),
Arrow::Table.load(@dir,
filter: ["equal", :visible, true]))
end
+
+ def test_schema
+ uri = build_file_uri(@path1)
+ @table1.save(uri)
+ schema = Arrow::Schema.new(visible: :boolean,
+ point: :int64)
+ assert_equal(Arrow::Table.new(schema,
+ [
+ @table1[:visible].data,
+ @table1[:point].cast(:int64),
+ ]),
+ Arrow::Table.load(uri, schema: schema))
+ end
end
end