| 20 | include Helper::Writable |
| 21 | |
| 22 | def setup |
| 23 | omit("Arrow Dataset is required") unless defined?(ArrowDataset) |
| 24 | Dir.mktmpdir do |tmpdir| |
| 25 | path = File.join(tmpdir, "table.arrow") |
| 26 | @table = build_table(visible: [ |
| 27 | build_boolean_array([true, false, true]), |
| 28 | build_boolean_array([false, true, false, true]), |
| 29 | ], |
| 30 | point: [ |
| 31 | build_int32_array([1, 2, 3]), |
| 32 | build_int32_array([-1, -2, -3, -4]), |
| 33 | ]) |
| 34 | @format = ArrowDataset::IPCFileFormat.new |
| 35 | write_table(@table, path) |
| 36 | factory = ArrowDataset::FileSystemDatasetFactory.new(@format) |
| 37 | factory.file_system_uri = build_file_uri(path) |
| 38 | @dataset = factory.finish |
| 39 | builder = @dataset.begin_scan |
| 40 | @scanner = builder.finish |
| 41 | yield |
| 42 | end |
| 43 | end |
| 44 | |
| 45 | def test_to_table |
| 46 | assert_equal(@table, @scanner.to_table) |
nothing calls this directly
no test coverage detected