mirror of
https://github.com/datahub-project/datahub.git
synced 2025-10-31 10:49:00 +00:00
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
import datahub.metadata.schema_classes as models
|
||
|
|
from datahub.emitter.mcp import MetadataChangeProposalWrapper
|
||
|
|
|
||
|
|
|
||
|
|
def test_mcpw_inference():
|
||
|
|
mcpw = MetadataChangeProposalWrapper(
|
||
|
|
entityUrn="urn:li:dataset:(urn:li:dataPlatform:bigquery,harshal-playground-306419.test_schema.excess_deaths_derived,PROD)",
|
||
|
|
aspect=models.DomainsClass(domains=["urn:li:domain:health"]),
|
||
|
|
)
|
||
|
|
assert mcpw.entityType == "dataset"
|
||
|
|
assert mcpw.aspectName == "domains"
|
||
|
|
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
mcpw = MetadataChangeProposalWrapper(
|
||
|
|
entityUrn="urn:li:dataset:(urn:li:dataPlatform:bigquery,harshal-playground-306419.test_schema.excess_deaths_derived,PROD)",
|
||
|
|
entityType="incorrect",
|
||
|
|
aspect=models.DomainsClass(domains=["urn:li:domain:health"]),
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_mcpw_from_obj():
|
||
|
|
# Checks that the MCPW from_obj() method returns a MCPW instead
|
||
|
|
# of an MCP with a serialized inner aspect object.
|
||
|
|
|
||
|
|
mcpw = MetadataChangeProposalWrapper(
|
||
|
|
entityUrn="urn:li:dataset:(urn:li:dataPlatform:bigquery,harshal-playground-306419.test_schema.excess_deaths_derived,PROD)",
|
||
|
|
aspect=models.DomainsClass(domains=["urn:li:domain:health"]),
|
||
|
|
)
|
||
|
|
|
||
|
|
mcpw2 = MetadataChangeProposalWrapper.from_obj(mcpw.to_obj())
|
||
|
|
|
||
|
|
assert isinstance(mcpw2, MetadataChangeProposalWrapper)
|
||
|
|
assert mcpw == mcpw2
|