parquetSchema more generic argument

This commit is contained in:
Kenny Daniel 2025-05-17 17:50:32 -07:00
parent 6b23a82894
commit c6bc226180
No known key found for this signature in database
GPG Key ID: 90AB653A8CAD7E45
3 changed files with 6 additions and 6 deletions

@ -50,7 +50,7 @@
"devDependencies": {
"@types/node": "22.15.18",
"@vitest/coverage-v8": "3.1.3",
"eslint": "9.26.0",
"eslint": "9.27.0",
"eslint-plugin-jsdoc": "50.6.17",
"hyparquet-compressors": "1.1.1",
"typescript": "5.8.3",

@ -187,11 +187,11 @@ export function parquetMetadata(arrayBuffer) {
/**
* Return a tree of schema elements from parquet metadata.
*
* @param {FileMetaData} metadata parquet metadata object
* @param {{schema: SchemaElement[]}} metadata parquet metadata object
* @returns {SchemaTree} tree of schema elements
*/
export function parquetSchema(metadata) {
return getSchemaPath(metadata.schema, [])[0]
export function parquetSchema({ schema }) {
return getSchemaPath(schema, [])[0]
}
/**

@ -6,8 +6,8 @@ describe('parquetSchema', () => {
it('parse schema tree from rowgroups.parquet', async () => {
const arrayBuffer = await asyncBufferFromFile('test/files/rowgroups.parquet')
const metadata = await parquetMetadataAsync(arrayBuffer)
const result = parquetSchema(metadata)
expect(result).toEqual(rowgroupsSchema)
const schemaTree = parquetSchema(metadata)
expect(schemaTree).toEqual(rowgroupsSchema)
})
})