def test_presto_get_fields(self): cols = [ { "name": "column" }, { "name": "column.nested_obj" }, { "name": 'column."quoted.nested obj"' }, ] actual_results = PrestoEngineSpec._get_fields(cols) expected_results = [ { "name": '"column"', "label": "column" }, { "name": '"column"."nested_obj"', "label": "column.nested_obj" }, { "name": '"column"."quoted.nested obj"', "label": 'column."quoted.nested obj"', }, ] for actual_result, expected_result in zip(actual_results, expected_results): self.assertEqual(actual_result.element.name, expected_result["name"]) self.assertEqual(actual_result.name, expected_result["label"])
def test_presto_get_fields(self): cols = [{ 'name': 'column' }, { 'name': 'column.nested_obj' }, { 'name': 'column."quoted.nested obj"' }] actual_results = PrestoEngineSpec._get_fields(cols) expected_results = [{ 'name': '"column"', 'label': 'column' }, { 'name': '"column"."nested_obj"', 'label': 'column.nested_obj' }, { 'name': '"column"."quoted.nested obj"', 'label': 'column."quoted.nested obj"' }] for actual_result, expected_result in zip(actual_results, expected_results): self.assertEqual(actual_result.element.name, expected_result['name']) self.assertEqual(actual_result.name, expected_result['label'])