mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-03 23:28:11 +00:00
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
# Copyright 2021 Acryl Data, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from datahub_actions.utils.collection_util import flatten_dict, flatten_list
|
|
|
|
|
|
def test_flatten_dict():
|
|
assert flatten_dict(
|
|
{
|
|
"entity": {
|
|
"urn": "urn:li:dataset:(urn:li:dataPlatform:hive,fct_users_created,PROD)",
|
|
"type": "DATASET",
|
|
},
|
|
"created": {
|
|
"time": 1645021517917,
|
|
"actor": {"urn": "urn:li:corpuser:admin"},
|
|
},
|
|
"urn": "urn:li:actionRequest:03c3fdc7-3147-48a1-af84-a1af1a45a202",
|
|
"type": "TERM_ASSOCIATION",
|
|
"params": {
|
|
"glossaryTermProposal": {
|
|
"glossaryTerm": {"urn": "urn:li:glossaryTerm:CustomerAccount"}
|
|
},
|
|
"tagProposal": None,
|
|
},
|
|
}
|
|
) == {
|
|
"entity_urn": "urn:li:dataset:(urn:li:dataPlatform:hive,fct_users_created,PROD)",
|
|
"entity_type": "DATASET",
|
|
"created_time": 1645021517917,
|
|
"created_actor_urn": "urn:li:corpuser:admin",
|
|
"urn": "urn:li:actionRequest:03c3fdc7-3147-48a1-af84-a1af1a45a202",
|
|
"type": "TERM_ASSOCIATION",
|
|
"params_glossaryTermProposal_glossaryTerm_urn": "urn:li:glossaryTerm:CustomerAccount",
|
|
"params_tagProposal": None,
|
|
}
|
|
|
|
|
|
def test_flatten_list():
|
|
assert flatten_list([["a"], ["b"], ["c"]]) == ["a", "b", "c"]
|