Python PrestoEngineSpec._filter_out_array_nested_cols Exemples

Exemple #1

Fichier : db_engine_specs_test.py Projet : Xaqt/incubator-superset

 def test_presto_filter_out_array_nested_cols(self):
     cols = [
         {"name": "column", "type": "ARRAY"},
         {"name": "column.nested_obj", "type": "FLOAT"},
     ]
     actual_filtered_cols, actual_array_cols = PrestoEngineSpec._filter_out_array_nested_cols(  # noqa ignore: E50
         cols
     )
     expected_filtered_cols = [{"name": "column", "type": "ARRAY"}]
     self.assertEqual(actual_filtered_cols, expected_filtered_cols)
     self.assertEqual(actual_array_cols, cols)

Exemple #2

 def test_presto_filter_out_array_nested_cols(self):
     cols = [{
         'name': 'column',
         'type': 'ARRAY'
     }, {
         'name': 'column.nested_obj',
         'type': 'FLOAT'
     }]
     actual_filtered_cols,\
         actual_array_cols = PrestoEngineSpec._filter_out_array_nested_cols(cols)
     expected_filtered_cols = [{'name': 'column', 'type': 'ARRAY'}]
     self.assertEqual(actual_filtered_cols, expected_filtered_cols)
     self.assertEqual(actual_array_cols, cols)