Python PrestoEngineSpec._expand_row_data Examples

Example #1

 def test_presto_expand_row_data(self):
     datum = {"row_col": [1, "a"]}
     row_column = "row_col"
     row_col_hierarchy = {
         "row_col": {
             "type": "ROW",
             "children": ["row_col.nested_int", "row_col.nested_str"],
         }
     }
     PrestoEngineSpec._expand_row_data(datum, row_column, row_col_hierarchy)
     expected_datum = {
         "row_col": [1, "a"],
         "row_col.nested_int": 1,
         "row_col.nested_str": "a",
     }
     self.assertEqual(datum, expected_datum)

Example #2

 def test_presto_expand_row_data(self):
     datum = {'row_col': [1, 'a']}
     row_column = 'row_col'
     row_col_hierarchy = {
         'row_col': {
             'type': 'ROW',
             'children': ['row_col.nested_int', 'row_col.nested_str'],
         },
     }
     PrestoEngineSpec._expand_row_data(datum, row_column, row_col_hierarchy)
     expected_datum = {
         'row_col': [1, 'a'],
         'row_col.nested_int': 1,
         'row_col.nested_str': 'a',
     }
     self.assertEqual(datum, expected_datum)