1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
// 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.
import {
generateRandomTables,
generateDictionaryTables
} from '../../../data/tables';
import { validateRecordBatchIterator } from '../validate';
import { Table, RecordBatchJSONWriter } from 'apache-arrow';
describe('RecordBatchJSONWriter', () => {
for (const table of generateRandomTables([10, 20, 30])) {
testJSONWriter(table, `[${table.schema.fields.join(', ')}]`);
}
for (const table of generateDictionaryTables([10, 20, 30])) {
testJSONWriter(table, `${table.schema.fields[0]}`);
}
});
function testJSONWriter(table: Table, name: string) {
describe(`should write the Arrow IPC JSON format (${name})`, () => {
test(`Table`, validateTable.bind(0, table));
});
}
async function validateTable(source: Table) {
const writer = RecordBatchJSONWriter.writeAll(source);
const result = Table.from(JSON.parse(await writer.toString()));
validateRecordBatchIterator(3, source.chunks);
expect(result).toEqualTable(source);
}
|
