2021-02-11 12:53:44 -08:00
|
|
|
import json
|
2021-02-11 23:14:20 -08:00
|
|
|
|
2021-02-11 12:53:44 -08:00
|
|
|
import deepdiff
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def load_json_file(filename: str) -> object:
|
|
|
|
|
with open(str(filename)) as f:
|
|
|
|
|
a = json.load(f)
|
|
|
|
|
return a
|
|
|
|
|
|
|
|
|
|
|
2021-04-14 13:40:24 -07:00
|
|
|
def assert_mces_equal(output: dict, golden: dict) -> None:
|
2021-02-11 12:53:44 -08:00
|
|
|
# This method assumes we're given a list of MCE json objects.
|
|
|
|
|
|
|
|
|
|
ignore_paths = {
|
|
|
|
|
# Ignore timestamps from the ETL pipeline. A couple examples:
|
|
|
|
|
# root[0]['proposedSnapshot']['com.linkedin.pegasus2avro.metadata.snapshot.DatasetSnapshot']['aspects'][0]['com.linkedin.pegasus2avro.common.Ownership']['lastModified']['time']
|
|
|
|
|
# root[69]['proposedSnapshot']['com.linkedin.pegasus2avro.metadata.snapshot.DatasetSnapshot']['aspects'][0]['com.linkedin.pegasus2avro.schema.SchemaMetadata']['lastModified']['time']"
|
|
|
|
|
r"root\[\d+\]\['proposedSnapshot'\].+\['aspects'\].+\['created'\]\['time'\]",
|
|
|
|
|
r"root\[\d+\]\['proposedSnapshot'\].+\['aspects'\].+\['lastModified'\]\['time'\]",
|
|
|
|
|
r"root\[\d+\]\['proposedSnapshot'\].+\['aspects'\].+\['createStamp'\]\['time'\]",
|
|
|
|
|
}
|
|
|
|
|
diff = deepdiff.DeepDiff(golden, output, exclude_regex_paths=ignore_paths)
|
|
|
|
|
assert not diff
|
