From e649a26c8c5ccea982c93aad923cdf0b74e119b5 Mon Sep 17 00:00:00 2001 From: Michal Swiatowy Date: Tue, 4 Jan 2022 11:59:19 +0100 Subject: [PATCH] fixed > 2 tblproperties parsing --- simple_ddl_parser/dialects/hql.py | 2 +- tests/test_hql_output_mode.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/simple_ddl_parser/dialects/hql.py b/simple_ddl_parser/dialects/hql.py index f0b5d46..426c869 100644 --- a/simple_ddl_parser/dialects/hql.py +++ b/simple_ddl_parser/dialects/hql.py @@ -44,7 +44,7 @@ def p_expression_tblproperties(self, p): def p_multi_assigments(self, p): """multi_assigments : LP assigment | multi_assigments RP - | multi_assigments COMMA assigment RP""" + | multi_assigments COMMA assigment""" p_list = remove_par(list(p)) p[0] = p_list[1] p[0].update(p_list[-1]) diff --git a/tests/test_hql_output_mode.py b/tests/test_hql_output_mode.py index 257c611..e4f3a22 100644 --- a/tests/test_hql_output_mode.py +++ b/tests/test_hql_output_mode.py @@ -1718,7 +1718,9 @@ def test_table_properties(): STORED AS PARQUET LOCATION 'hdfs://test' TBLPROPERTIES ( 'parquet.compression'='SNAPPY', - 'parquet.compression2'='SNAPPY2' + 'parquet.compression2'='SNAPPY2', + 'parquet.compression3'='SNAPPY3', + 'parquet.compression4'='SNAPPY4' ) """ result = DDLParser(ddl).run(group_by_type=True, output_mode="hql") @@ -1762,6 +1764,8 @@ def test_table_properties(): "tblproperties": { "'parquet.compression'": "'SNAPPY'", "'parquet.compression2'": "'SNAPPY2'", + "'parquet.compression3'": "'SNAPPY3'", + "'parquet.compression4'": "'SNAPPY4'", }, } ],