Merge pull request ironSource#30 from Aletheios/master

Added ENUM to types
jeffbski · Sep 7, 2018 · 9e123a3 · 9e123a3
2 parents f19fd63 + 4c4f179
commit 9e123a3
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/lib/types.js b/lib/types.js
@@ -41,6 +41,12 @@ const PARQUET_LOGICAL_TYPES = {
     toPrimitive: toPrimitive_UTF8,
     fromPrimitive: fromPrimitive_UTF8
   },
+  'ENUM': {
+    primitiveType: 'BYTE_ARRAY',
+    originalType: 'UTF8',
+    toPrimitive: toPrimitive_UTF8,
+    fromPrimitive: fromPrimitive_UTF8
+  },
   'TIME_MILLIS': {
     primitiveType: 'INT32',
     originalType: 'TIME_MILLIS',

diff --git a/test/test-files.js b/test/test-files.js
@@ -108,6 +108,11 @@ describe('test-files', function() {
     assert.deepEqual(data,[{foo: 'bar'},{}]);
   });
 
+  it('test-enum-type.parquet loads', async function() {
+    const data = await readData('test-enum-type.parquet');
+    assert.deepEqual(data,[{foo: 'bar'}]);
+  });
+
   it('test-null-dictionary.parquet loads', async function() {
     const data = await readData('test-null-dictionary.parquet');
     assert.deepEqual(data,[].concat.apply([{}],[...Array(3)].map( () => ([{foo: 'bar'}, {foo: 'baz'}]))));

diff --git a/test/test-files/test-enum-type.parquet b/test/test-files/test-enum-type.parquet